summaryrefslogtreecommitdiff
path: root/text_recognizer/networks
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks')
-rw-r--r--text_recognizer/networks/encoders/efficientnet/efficientnet.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
index 32934d5..8c1ebcf 100644
--- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py
+++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
@@ -101,7 +101,7 @@ class EfficientNet(nn.Module):
momentum=self.bn_momentum,
eps=self.bn_eps,
),
- nn.Dropout(p=self.arch[-1]),
+ nn.Dropout(p=self.params[-1]),
)
def extract_features(self, x: Tensor) -> Tensor: