diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:58:00 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:58:00 +0200 |
| commit | 687e3c829d1b50a2f3bf96bafc0b29cff02e30d4 (patch) | |
| tree | 31769dbb8be1cb3e789b1a38872af00e9d291fe7 /text_recognizer/networks/efficientnet | |
| parent | b76e27e895405dbaaf50c828817a71c9ec939de0 (diff) | |
Update block args for efficient net
Diffstat (limited to 'text_recognizer/networks/efficientnet')
| -rw-r--r-- | text_recognizer/networks/efficientnet/utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/efficientnet/utils.py b/text_recognizer/networks/efficientnet/utils.py index 770f85b..e5b6433 100644 --- a/text_recognizer/networks/efficientnet/utils.py +++ b/text_recognizer/networks/efficientnet/utils.py @@ -77,7 +77,7 @@ def block_args() -> List[DictConfig]: [2, 5, (2, 2), 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, (2, 1), 6, 112, 192, 0.25], + [4, 5, (1, 1), 6, 112, 192, 0.25], [1, 3, (2, 1), 6, 192, 320, 0.25], ] block_args_ = [] |