diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-08 12:41:04 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-08 12:41:04 +0100 |
| commit | beeaef529e7c893a3475fe27edc880e283373725 (patch) | |
| tree | 59eb72562bf7a5a9470c2586e6280600ad94f1ae /src/text_recognizer/networks/ctc.py | |
| parent | 4d7713746eb936832e84852e90292936b933e87d (diff) | |
Trying to get the CNNTransformer to work, but it is hard.
Diffstat (limited to 'src/text_recognizer/networks/ctc.py')
| -rw-r--r-- | src/text_recognizer/networks/ctc.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/text_recognizer/networks/ctc.py b/src/text_recognizer/networks/ctc.py index 2493d5c..af9b700 100644 --- a/src/text_recognizer/networks/ctc.py +++ b/src/text_recognizer/networks/ctc.py @@ -33,7 +33,7 @@ def greedy_decoder( """ if character_mapper is None: - character_mapper = EmnistMapper() + character_mapper = EmnistMapper(pad_token="_") # noqa: S106 predictions = rearrange(torch.argmax(predictions, dim=2), "t b -> b t") decoded_predictions = [] |