From 4c5c5abfb6cb214d246a9c1e19916dd0ebd4e2b6 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 7 Nov 2021 23:49:20 +0100
Subject: Fix dropout in efficientnet

---
 text_recognizer/networks/encoders/efficientnet/efficientnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'text_recognizer/networks/encoders/efficientnet')

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:
-- 
cgit v1.2.3-70-g09d2