diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:24:20 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:24:20 +0200 |
| commit | dedf8deb025ac9efdad5e9baf9165ef63d6829ff (patch) | |
| tree | 56b10fcaef479d8abe9b0e6c05e07ad5e02b9ab0 /text_recognizer/networks/transducer/transducer.py | |
| parent | 532286b516b17d279c321358bf03dddc8adc8029 (diff) | |
Pre-commit fixes, optimizer loading fix
Diffstat (limited to 'text_recognizer/networks/transducer/transducer.py')
| -rw-r--r-- | text_recognizer/networks/transducer/transducer.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/text_recognizer/networks/transducer/transducer.py b/text_recognizer/networks/transducer/transducer.py index d7e3d08..b10f93a 100644 --- a/text_recognizer/networks/transducer/transducer.py +++ b/text_recognizer/networks/transducer/transducer.py @@ -392,7 +392,12 @@ def load_transducer_loss( transitions = gtn.load(str(processed_path / transitions)) preprocessor = Preprocessor( - data_dir, num_features, tokens_path, lexicon_path, use_words, prepend_wordsep, + data_dir, + num_features, + tokens_path, + lexicon_path, + use_words, + prepend_wordsep, ) num_tokens = preprocessor.num_tokens |