From 64189f0790a78e2daaa6a47c779add9a21ef5814 Mon Sep 17 00:00:00 2001
From: guarin <43336610+guarin@users.noreply.github.com>
Date: Tue, 30 Jan 2024 09:47:05 +0100
Subject: [PATCH] Revert "Fix TiCoLoss"

---
 lightly/loss/tico_loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lightly/loss/tico_loss.py b/lightly/loss/tico_loss.py
index b2ae3f328..1528a3ffc 100644
--- a/lightly/loss/tico_loss.py
+++ b/lightly/loss/tico_loss.py
@@ -101,7 +101,7 @@ def forward(
         z_b = torch.nn.functional.normalize(z_b, dim=1)
 
         # compute auxiliary matrix B
-        B = torch.mm(z_a.T, z_a).detach() / z_a.shape[0]
+        B = torch.mm(z_a.T, z_a) / z_a.shape[0]
 
         # init covariance matrix
         if self.C is None: