diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-19 19:55:59 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-19 19:55:59 +0200 |
commit | a347b7e08df174d10d2cdffc269f42a2a03490a2 (patch) | |
tree | eeabed26f750064f83207fddc2786fd66d33ddeb /training/conf/network | |
parent | 3cac3c32ff49618cfa05bb4e5630a62fc31af40a (diff) |
Update configs
Diffstat (limited to 'training/conf/network')
-rw-r--r-- | training/conf/network/conv_transformer.yaml | 3 | ||||
-rw-r--r-- | training/conf/network/efficientnet.yaml | 7 |
2 files changed, 9 insertions, 1 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index 39c5c46..ccdf960 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -11,6 +11,7 @@ encoder: bn_eps: 1.0e-3 depth: 3 out_channels: *hidden_dim + stride: [2, 1] decoder: _target_: text_recognizer.networks.transformer.Decoder depth: 6 @@ -46,4 +47,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [72, 80] + shape: [17, 78] 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 |