diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:54 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:54 +0200 |
commit | b69254ce3135c112e29f7f1c986b7f0817da0c33 (patch) | |
tree | e596a719b445c0cbbd4108079206ec9d14de1437 /training/conf/experiment/cnn_htr_char_lines.yaml | |
parent | cba94bdcab90f288dd1172607500ba2b28279736 (diff) |
Update configs
Diffstat (limited to 'training/conf/experiment/cnn_htr_char_lines.yaml')
-rw-r--r-- | training/conf/experiment/cnn_htr_char_lines.yaml | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/training/conf/experiment/cnn_htr_char_lines.yaml b/training/conf/experiment/cnn_htr_char_lines.yaml index 0d62a73..53f6d91 100644 --- a/training/conf/experiment/cnn_htr_char_lines.yaml +++ b/training/conf/experiment/cnn_htr_char_lines.yaml @@ -1,5 +1,3 @@ -# @package _global_ - defaults: - override /mapping: null - override /criterion: null @@ -19,11 +17,10 @@ criterion: _target_: text_recognizer.criterions.label_smoothing.LabelSmoothingLoss smoothing: 0.1 ignore_index: *ignore_index - # _target_: torch.nn.CrossEntropyLoss - # ignore_index: *ignore_index mapping: - _target_: text_recognizer.data.emnist_mapping.EmnistMapping + mapping: &mapping + _target_: text_recognizer.data.emnist_mapping.EmnistMapping callbacks: stochastic_weight_averaging: @@ -73,6 +70,7 @@ datamodule: augment: true pin_memory: true word_pieces: false + <<: *mapping network: _target_: text_recognizer.networks.conv_transformer.ConvTransformer @@ -80,6 +78,7 @@ network: hidden_dim: &hidden_dim 128 encoder_dim: 1280 dropout_rate: 0.2 + <<: *mapping num_classes: *num_classes pad_index: *ignore_index encoder: |