print

alicebizeul · Apr 16, 2023 · 2586e34 · 2586e34
1 parent 6a64f89
commit 2586e34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simclr/modules/custom_infonce.py b/simclr/modules/custom_infonce.py
@@ -41,7 +41,7 @@ def forward(self, anchor_rec, positive_rec):
  # total_loss_value = torch.mean(- pos + neg)
  if self.bound:
  if self.subsample:
- keep=random.shuffle(list(np.arange(anchor_rec.shape[1])))
+ keep=random.shuffle(list(np.arange(int(anchor_rec.shape[1]))))
  print(keep) #[:int(int(anchor_rec.shape[1])/2)]
  sim11=sim11[:,keep,:]
  sim12=sim12[:,keep,:]