diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-04 18:21:13 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-04 18:21:13 +0200 |
| commit | 03dae09c63f2079f37bbf25fd9ded6f20f1490da (patch) | |
| tree | 0ce80f3a0c8fafaea2e3d2734e4ddc577c504b2d /text_recognizer/models/base.py | |
| parent | 8207dc902db439d606fe36d726f0405aabbf173e (diff) | |
Add transformer lit model
Diffstat (limited to 'text_recognizer/models/base.py')
| -rw-r--r-- | text_recognizer/models/base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 0c70625..46e5136 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -11,7 +11,7 @@ import torchmetrics from text_recognizer import networks -class BaseModel(pl.LightningModule): +class LitBaseModel(pl.LightningModule): """Abstract PyTorch Lightning class.""" def __init__( |