diff options
Diffstat (limited to 'training/conf')
| -rw-r--r-- | training/conf/datamodule/emnist_lines.yaml | 2 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 4 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 4 | ||||
| -rw-r--r-- | training/conf/experiment/conv_transformer_paragraphs.yaml | 3 | ||||
| -rw-r--r-- | training/conf/mapping/characters.yaml | 2 | ||||
| -rw-r--r-- | training/conf/tokenizer/default.yaml | 2 |
6 files changed, 9 insertions, 8 deletions
diff --git a/training/conf/datamodule/emnist_lines.yaml b/training/conf/datamodule/emnist_lines.yaml index 218df6c..ce35c3e 100644 --- a/training/conf/datamodule/emnist_lines.yaml +++ b/training/conf/datamodule/emnist_lines.yaml @@ -6,4 +6,4 @@ pin_memory: true transform: transform/lines.yaml test_transform: test_transform/lines.yaml mapping: - _target_: text_recognizer.data.mappings.EmnistMapping + _target_: text_recognizer.data.tokenizer.Tokenizer diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index c46714c..64c3964 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -13,6 +13,6 @@ target_transform: _target_: text_recognizer.data.transforms.pad.Pad max_len: 682 pad_index: 3 -mapping: - _target_: text_recognizer.data.mappings.EmnistMapping +tokenizer: + _target_: text_recognizer.data.tokenizer.Tokenizer extra_symbols: ["\n"] diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index 4f1f1b8..f84116d 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -9,5 +9,5 @@ transform: test_transform: _target_: text_recognizer.data.stems.line.IamLinesStem augment: false -mapping: - _target_: text_recognizer.data.mappings.EmnistMapping +tokenizer: + _target_: text_recognizer.data.tokenizer.Tokenizer diff --git a/training/conf/experiment/conv_transformer_paragraphs.yaml b/training/conf/experiment/conv_transformer_paragraphs.yaml index 60ff1bf..cdac387 100644 --- a/training/conf/experiment/conv_transformer_paragraphs.yaml +++ b/training/conf/experiment/conv_transformer_paragraphs.yaml @@ -84,8 +84,9 @@ network: decoder: _target_: text_recognizer.networks.transformer.Decoder depth: 6 + dim: *hidden_dim block: - _target_: text_recognizer.networks.transformer.DecoderBlock + _target_: text_recognizer.networks.transformer.decoder_block.DecoderBlock self_attn: _target_: text_recognizer.networks.transformer.Attention dim: *hidden_dim diff --git a/training/conf/mapping/characters.yaml b/training/conf/mapping/characters.yaml deleted file mode 100644 index 8cbd55d..0000000 --- a/training/conf/mapping/characters.yaml +++ /dev/null @@ -1,2 +0,0 @@ -_target_: text_recognizer.data.mappings.EmnistMapping -extra_symbols: [ "\n" ] diff --git a/training/conf/tokenizer/default.yaml b/training/conf/tokenizer/default.yaml new file mode 100644 index 0000000..2b1a8c9 --- /dev/null +++ b/training/conf/tokenizer/default.yaml @@ -0,0 +1,2 @@ +_target_: text_recognizer.data.tokenizer.Tokenizer +extra_symbols: ["\n"] |