summaryrefslogtreecommitdiff
path: root/training/conf/network
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-29 23:59:52 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-29 23:59:52 +0200
commit34098ccbbbf6379c0bd29a987440b8479c743746 (patch)
treea8c68e3036503049fc7034c677ec855465f7a8e0 /training/conf/network
parentc032ffb05a7ed86f8fe5d596f94e8997c558cae8 (diff)
Configs, refactor with attrs, fix attr bug in iam
Diffstat (limited to 'training/conf/network')
-rw-r--r--training/conf/network/conv_transformer.yaml13
1 files changed, 13 insertions, 0 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml
new file mode 100644
index 0000000..f72e030
--- /dev/null
+++ b/training/conf/network/conv_transformer.yaml
@@ -0,0 +1,13 @@
+defaults:
+ - encoder: efficientnet
+ - decoder: transformer_decoder
+
+_target_: text_recognizer.networks.conv_transformer.ConvTransformer
+input_dims: [1, 576, 640]
+hidden_dim: 256
+dropout_rate: 0.2
+max_output_len: 682
+num_classes: 1004
+start_token: <s>
+end_token: <e>
+pad_token: <p>