From 687e3c829d1b50a2f3bf96bafc0b29cff02e30d4 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 20 Jun 2022 00:58:00 +0200
Subject: Update block args for efficient net
---
text_recognizer/networks/efficientnet/utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
(limited to 'text_recognizer')
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_ = []
--
cgit v1.2.3-70-g09d2