summaryrefslogtreecommitdiff
path: root/training/conf/network/efficientnet.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-19 19:55:59 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-19 19:55:59 +0200
commita347b7e08df174d10d2cdffc269f42a2a03490a2 (patch)
treeeeabed26f750064f83207fddc2786fd66d33ddeb /training/conf/network/efficientnet.yaml
parent3cac3c32ff49618cfa05bb4e5630a62fc31af40a (diff)
Update configs
Diffstat (limited to 'training/conf/network/efficientnet.yaml')
-rw-r--r--training/conf/network/efficientnet.yaml7
1 files changed, 7 insertions, 0 deletions
diff --git a/training/conf/network/efficientnet.yaml b/training/conf/network/efficientnet.yaml
new file mode 100644
index 0000000..bbe68dd
--- /dev/null
+++ b/training/conf/network/efficientnet.yaml
@@ -0,0 +1,7 @@
+_target_: text_recognizer.networks.efficientnet.EfficientNet
+arch: b0
+stochastic_dropout_rate: 0.2
+bn_momentum: 0.99
+bn_eps: 1.0e-3
+stride: [2, 1]
+out_channels: 144