diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-22 22:27:17 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-22 22:27:17 +0200 |
| commit | a30acba378f6f8fb86bb1a6096b9eb0ae7a68449 (patch) | |
| tree | c4a1087520a05a1f0bc792887b1ab51694a59b84 /training/conf/network | |
| parent | 1d592a432e57eadff0b94f945e59cbeb95e6a3f0 (diff) | |
Update configs
Diffstat (limited to 'training/conf/network')
| -rw-r--r-- | training/conf/network/conv_transformer.yaml | 4 | ||||
| -rw-r--r-- | training/conf/network/efficientnet.yaml | 1 |
2 files changed, 2 insertions, 3 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index 016416c..24857c8 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 128 +hidden_dim: &hidden_dim 144 num_classes: 58 pad_index: 3 encoder: @@ -46,4 +46,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [18, 80] + shape: [18, 79] diff --git a/training/conf/network/efficientnet.yaml b/training/conf/network/efficientnet.yaml index bbe68dd..4dd148f 100644 --- a/training/conf/network/efficientnet.yaml +++ b/training/conf/network/efficientnet.yaml @@ -3,5 +3,4 @@ arch: b0 stochastic_dropout_rate: 0.2 bn_momentum: 0.99 bn_eps: 1.0e-3 -stride: [2, 1] out_channels: 144 |