diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:51:59 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:51:59 +0200 |
| commit | 04876c35a1eeea68794745f1ca113754c0d97757 (patch) | |
| tree | b4664a71c031a9ced6f60c9e72632bad817f11df /text_recognizer/models/transformer.py | |
| parent | 7268035fb9e57342612a8cc50a1fe04e8841ca2f (diff) | |
black reformatting
Diffstat (limited to 'text_recognizer/models/transformer.py')
| -rw-r--r-- | text_recognizer/models/transformer.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/models/transformer.py b/text_recognizer/models/transformer.py index 7a9d566..0e01bb5 100644 --- a/text_recognizer/models/transformer.py +++ b/text_recognizer/models/transformer.py @@ -13,6 +13,7 @@ from text_recognizer.models.base import BaseLitModel @attr.s(auto_attribs=True) class TransformerLitModel(BaseLitModel): """A PyTorch Lightning model for transformer networks.""" + mapping: Type[AbstractMapping] = attr.ib() start_token: str = attr.ib() end_token: str = attr.ib() |