From b69254ce3135c112e29f7f1c986b7f0817da0c33 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 10 Oct 2021 18:06:54 +0200 Subject: Update configs --- training/conf/experiment/cnn_htr_char_lines.yaml | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'training/conf/experiment/cnn_htr_char_lines.yaml') 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: -- cgit v1.2.3-70-g09d2