diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-25 22:31:36 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-25 22:31:36 +0200 |
| commit | da29c1cf4d062087f1b29dc9402ee6384203b690 (patch) | |
| tree | a19b035990864a0be3d23faf3cf21382d60c5eda /text_recognizer/data/transforms | |
| parent | 00dd3df9f2e29622248668662cb40ff0c8889145 (diff) | |
Make data dir static in preprocessor and word piece
Diffstat (limited to 'text_recognizer/data/transforms')
| -rw-r--r-- | text_recognizer/data/transforms/word_piece.py | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/text_recognizer/data/transforms/word_piece.py b/text_recognizer/data/transforms/word_piece.py index 6bf5472..69f0ce1 100644 --- a/text_recognizer/data/transforms/word_piece.py +++ b/text_recognizer/data/transforms/word_piece.py @@ -16,7 +16,6 @@ class WordPiece: num_features: int = 1000, tokens: str = "iamdb_1kwp_tokens_1000.txt", lexicon: str = "iamdb_1kwp_lex_1000.txt", - data_dir: Optional[Union[str, Path]] = None, use_words: bool = False, prepend_wordsep: bool = False, special_tokens: Set[str] = {"<s>", "<e>", "<p>"}, @@ -24,7 +23,6 @@ class WordPiece: max_len: int = 451, ) -> None: self.mapping = WordPieceMapping( - data_dir=data_dir, num_features=num_features, tokens=tokens, lexicon=lexicon, |