summaryrefslogtreecommitdiff
path: root/text_recognizer/models
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-24 00:57:00 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-24 00:57:00 +0200
commit18af813a7a736d460b902e0856b2b17a9a67703b (patch)
treee2f6fd9fc0b73dbd0ecf5b11d7d0f53095d8320c /text_recognizer/models
parentc7475336442b767182afa5d2c63a02313e1a9a12 (diff)
Format of files
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."""