From 18af813a7a736d460b902e0856b2b17a9a67703b Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 24 Oct 2021 00:57:00 +0200 Subject: Format of files --- text_recognizer/models/base.py | 1 + text_recognizer/networks/barlow_twins/network.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer') 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 - -- cgit v1.2.3-70-g09d2