diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-24 00:57:00 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-24 00:57:00 +0200 |
| commit | 18af813a7a736d460b902e0856b2b17a9a67703b (patch) | |
| tree | e2f6fd9fc0b73dbd0ecf5b11d7d0f53095d8320c /text_recognizer | |
| parent | c7475336442b767182afa5d2c63a02313e1a9a12 (diff) | |
Format of files
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/models/base.py | 1 | ||||
| -rw-r--r-- | text_recognizer/networks/barlow_twins/network.py | 1 |
2 files changed, 1 insertions, 1 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.""" diff --git a/text_recognizer/networks/barlow_twins/network.py b/text_recognizer/networks/barlow_twins/network.py index 874e570..a3e3750 100644 --- a/text_recognizer/networks/barlow_twins/network.py +++ b/text_recognizer/networks/barlow_twins/network.py @@ -16,4 +16,3 @@ class BarlowTwins(nn.Module): z_e = F.adaptive_avg_pool2d(z, (1, 1)).flatten(start_dim=1) z_p = self.projector(z_e) return z_p - |