diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:56:50 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:56:50 +0200 |
| commit | 598fd271942884f852e98c82d6caedad425813a1 (patch) | |
| tree | 19adb7c3960ecb4079cb02a8ee38812caee7a222 /training | |
| parent | 892137c4f1104edc9645c0e1e877b885816fa83c (diff) | |
Remove strides
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/network/conv_transformer.yaml | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index 90c2cb8..016416c 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -1,6 +1,6 @@ _target_: text_recognizer.networks.ConvTransformer input_dims: [1, 1, 576, 640] -hidden_dim: &hidden_dim 144 +hidden_dim: &hidden_dim 128 num_classes: 58 pad_index: 3 encoder: @@ -11,7 +11,6 @@ encoder: bn_eps: 1.0e-3 depth: 5 out_channels: *hidden_dim - stride: [2, 1] decoder: _target_: text_recognizer.networks.transformer.Decoder depth: 6 @@ -47,4 +46,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [18, 78] + shape: [18, 80] |