diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-07 23:49:20 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-07 23:49:20 +0100 |
| commit | 4c5c5abfb6cb214d246a9c1e19916dd0ebd4e2b6 (patch) | |
| tree | 37bf50ca5453d98f7142b5bb90baa4a37de41d1a | |
| parent | 8728fc193af64de1eb70c0b24b1c3e80d97073dc (diff) | |
Fix dropout in efficientnet
| -rw-r--r-- | text_recognizer/networks/encoders/efficientnet/efficientnet.py | 2 |
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: |