diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-25 23:32:50 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-25 23:32:50 +0200 |
| commit | 9426cc794d8c28a65bbbf5ae5466a0a343078558 (patch) | |
| tree | 44e31b0a7c58597d603ac29a693462aae4b6e9b0 /text_recognizer/networks/transducer | |
| parent | 4e60c836fb710baceba570c28c06437db3ad5c9b (diff) | |
Efficient net and non working transformer model.
Diffstat (limited to 'text_recognizer/networks/transducer')
| -rw-r--r-- | text_recognizer/networks/transducer/transducer.py | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/text_recognizer/networks/transducer/transducer.py b/text_recognizer/networks/transducer/transducer.py index b10f93a..d7e3d08 100644 --- a/text_recognizer/networks/transducer/transducer.py +++ b/text_recognizer/networks/transducer/transducer.py @@ -392,12 +392,7 @@ 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 |