diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:09:20 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-20 00:09:20 +0200 |
| commit | 521f3bbbba9f04f48e81d78033c6e1c29a08e515 (patch) | |
| tree | 1ffb50acde358fb151d114f63b760e67c77e3274 /training/conf/network/conv_transformer.yaml | |
| parent | ce3f63801013aba2f05cfb92f1a3a87393610d27 (diff) | |
Update eff net config
Diffstat (limited to 'training/conf/network/conv_transformer.yaml')
| -rw-r--r-- | training/conf/network/conv_transformer.yaml | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml index ccdf960..90c2cb8 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -9,7 +9,7 @@ encoder: stochastic_dropout_rate: 0.2 bn_momentum: 0.99 bn_eps: 1.0e-3 - depth: 3 + depth: 5 out_channels: *hidden_dim stride: [2, 1] decoder: @@ -47,4 +47,4 @@ decoder: pixel_embedding: _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding dim: *hidden_dim - shape: [17, 78] + shape: [18, 78] |