summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/encoders/efficientnet/efficientnet.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-08 22:25:04 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-08 22:25:04 +0200
commit53bfdaa0b4d3a04c5f2a274c5657ada81e9bf135 (patch)
tree2509e7bb5a25311788d654e3a2e665578f9c9403 /text_recognizer/networks/encoders/efficientnet/efficientnet.py
parentd717f9e3e4dd17351f08c5822cb90d055c4513cc (diff)
Add comments
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/efficientnet.py')
-rw-r--r--text_recognizer/networks/encoders/efficientnet/efficientnet.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
index fb4f002..59598b5 100644
--- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py
+++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py
@@ -10,6 +10,7 @@ from .utils import (
class EfficientNet(nn.Module):
+ # TODO: attr
archs = {
# width,depth0res,dropout
"b0": (1.0, 1.0, 0.2),