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