diff options
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: |