diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:09:20 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:09:20 +0200 |
| commit | 521f3bbbba9f04f48e81d78033c6e1c29a08e515 (patch) | |
| tree | 1ffb50acde358fb151d114f63b760e67c77e3274 /text_recognizer/networks/efficientnet/utils.py | |
| parent | ce3f63801013aba2f05cfb92f1a3a87393610d27 (diff) | |
Update eff net config
Diffstat (limited to 'text_recognizer/networks/efficientnet/utils.py')
| -rw-r--r-- | text_recognizer/networks/efficientnet/utils.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/text_recognizer/networks/efficientnet/utils.py b/text_recognizer/networks/efficientnet/utils.py index 5234324..770f85b 100644 --- a/text_recognizer/networks/efficientnet/utils.py +++ b/text_recognizer/networks/efficientnet/utils.py @@ -72,13 +72,13 @@ def block_args() -> List[DictConfig]: "se_ratio", ] args = [ - [1, 3, (1, 1), 1, 32, 16, 0.25], + [1, 3, (1, 1), 1, 16, 16, 0.25], [2, 3, (2, 2), 6, 16, 24, 0.25], [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], + [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], + [1, 3, (2, 1), 6, 192, 320, 0.25], ] block_args_ = [] for row in args: |