print

alicebizeul · Apr 14, 2023 · 1544cc3 · 1544cc3
1 parent abff38e
commit 1544cc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simclr/modules/custom_infonce.py b/simclr/modules/custom_infonce.py
@@ -43,7 +43,7 @@ def forward(self, anchor_rec, positive_rec):
 
  print("After",sim11)
 
- print("Removal of duplo",sim11.shape,sim12[..., range(d), range(d)])
+ print("Removal of duplo",sim11.shape,)
 
  # if not self.simclr_compatibility:
  # pos = sim12[..., range(d), range(d)]