diff options
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__( |