summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-17 22:42:15 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-17 22:42:15 +0100
commit91db5e23f86ec0b829aebef6eef642bcf63da53b (patch)
tree45c9055139f4c59244e61541021d0c017f9f6f2c /text_recognizer
parentc28126f92672c89ebdb9fc150bc16667f70f5515 (diff)
Replace last two layers in efficientnet with one
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/encoders/efficientnet/utils.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/text_recognizer/networks/encoders/efficientnet/utils.py b/text_recognizer/networks/encoders/efficientnet/utils.py
index 5234324..2b1aebb 100644
--- a/text_recognizer/networks/encoders/efficientnet/utils.py
+++ b/text_recognizer/networks/encoders/efficientnet/utils.py
@@ -77,8 +77,7 @@ def block_args() -> List[DictConfig]:
[2, 5, (2, 2), 6, 24, 40, 0.25],
[3, 3, (2, 2), 6, 40, 80, 0.25],
[3, 5, (1, 1), 6, 80, 112, 0.25],
- [4, 5, (2, 2), 6, 112, 192, 0.25],
- [1, 3, (1, 1), 6, 192, 320, 0.25],
+ [1, 3, (1, 1), 6, 112, 320, 0.25],
]
block_args_ = []
for row in args: