summaryrefslogtreecommitdiff
path: root/text_recognizer/networks
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks')
-rw-r--r--text_recognizer/networks/efficientnet/utils.py4
1 files changed, 2 insertions, 2 deletions
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],