From 91db5e23f86ec0b829aebef6eef642bcf63da53b Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 17 Nov 2021 22:42:15 +0100 Subject: Replace last two layers in efficientnet with one --- text_recognizer/networks/encoders/efficientnet/utils.py | 3 +-- 1 file changed, 1 insertion(+), 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: -- cgit v1.2.3-70-g09d2