Update misc.py

fix inplace operation (pytorch bug)
jankrepl · Dec 25, 2023 · 946fd37 · 946fd37
1 parent 341b34d
commit 946fd37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepdow/layers/misc.py b/deepdow/layers/misc.py
@@ -141,7 +141,7 @@ def compute_covariance(m, shrinkage_strategy=None, shrinkage_coef=0.5):
 
         """
         fact = 1.0 / (m.size(1) - 1)
-        m -= torch.mean(m, dim=1, keepdim=True)  # !!!!!!!!!!! INPLACE
+        m = m - torch.mean(m, dim=1, keepdim=True)  # !!!!!!!!!!! INPLACE
         mt = m.t()
 
         s = fact * m.matmul(mt)  # sample covariance matrix