From 1d592a432e57eadff0b94f945e59cbeb95e6a3f0 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 22 Jun 2022 22:26:59 +0200 Subject: Update efficient net config --- text_recognizer/networks/efficientnet/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'text_recognizer/networks/efficientnet/utils.py') diff --git a/text_recognizer/networks/efficientnet/utils.py b/text_recognizer/networks/efficientnet/utils.py index e5b6433..412d07d 100644 --- a/text_recognizer/networks/efficientnet/utils.py +++ b/text_recognizer/networks/efficientnet/utils.py @@ -72,9 +72,9 @@ def block_args() -> List[DictConfig]: "se_ratio", ] args = [ - [1, 3, (1, 1), 1, 16, 16, 0.25], + [1, 3, (1, 1), 1, 32, 16, 0.25], [2, 3, (2, 2), 6, 16, 24, 0.25], - [2, 5, (2, 2), 6, 24, 40, 0.25], + [2, 5, (2, 1), 6, 24, 40, 0.25], [3, 3, (2, 1), 6, 40, 80, 0.25], [3, 5, (2, 1), 6, 80, 112, 0.25], [4, 5, (1, 1), 6, 112, 192, 0.25], -- cgit v1.2.3-70-g09d2