summaryrefslogtreecommitdiff
path: root/text_recognizer/models
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/models')
-rw-r--r--text_recognizer/models/base.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py
index 8aadc39..ea2a17a 100644
--- a/text_recognizer/models/base.py
+++ b/text_recognizer/models/base.py
@@ -13,6 +13,7 @@ import torchmetrics
from text_recognizer.data.mappings.base_mapping import AbstractMapping
+
@attr.s(eq=False)
class BaseLitModel(LightningModule):
"""Abstract PyTorch Lightning class."""