diff options
Diffstat (limited to 'text_recognizer/networks/barlow_twins')
| -rw-r--r-- | text_recognizer/networks/barlow_twins/network.py | 1 |
1 files changed, 0 insertions, 1 deletions
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 - |