diff options
Diffstat (limited to 'training/conf/experiment')
| -rw-r--r-- | training/conf/experiment/conv_transformer_lines.yaml | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/training/conf/experiment/conv_transformer_lines.yaml b/training/conf/experiment/conv_transformer_lines.yaml index 6ba4535..11646ca 100644 --- a/training/conf/experiment/conv_transformer_lines.yaml +++ b/training/conf/experiment/conv_transformer_lines.yaml @@ -10,7 +10,7 @@ defaults: - override /lr_schedulers: null - override /optimizers: null -epochs: &epochs 256 +epochs: &epochs 620 ignore_index: &ignore_index 3 num_classes: &num_classes 57 max_output_len: &max_output_len 89 @@ -44,7 +44,7 @@ optimizers: lr_schedulers: network: _target_: torch.optim.lr_scheduler.CosineAnnealingLR - T_max: 256 + T_max: *epochs eta_min: 1.0e-5 last_epoch: -1 interval: epoch @@ -64,7 +64,7 @@ rotary_embedding: &rotary_embedding attn: &attn dim: &hidden_dim 256 - num_heads: 8 + num_heads: 4 dim_head: 64 dropout_rate: &dropout_rate 0.5 @@ -76,7 +76,7 @@ network: pad_index: *ignore_index encoder: _target_: text_recognizer.networks.encoders.efficientnet.EfficientNet - arch: b2 + arch: b3 stochastic_dropout_rate: 0.2 bn_momentum: 0.99 bn_eps: 1.0e-3 |