fix : negative pair is always negative

image_ref/dataset_ref.py

@@ -133,7 +133,7 @@ class ImageFolderDuo(data.Dataset):
             if i < self.positive_prop:
                 label_ref = target
             else :
-                l = np.arange(0,len(self.classes)-1,dtype=int).tolist()
+                l = np.arange(0,len(self.classes),dtype=int).tolist()
                 l.remove(target)
                 label_ref = np.random.choice(l)
         else :