fix : typo out file name

image_ref/dataset_ref.py

@@ -145,8 +145,8 @@ class ImageFolderDuo(data.Dataset):
             imgAER = self.transform(imgAER)
             imgANA = self.transform(imgANA)
             img_ref = self.transform(img_ref)
-        target = 0 if target == label_ref else 1
-        return imgAER, imgANA, img_ref, target
+        contrastive_target = 0 if target == label_ref else 1
+        return imgAER, imgANA, img_ref, contrastive_target
 
     def __len__(self):
         return len(self.imlist)

image_ref/main.py

@@ -133,7 +133,7 @@ def run_duo(args):
 
 
     plt.show()
-    plt.savefig('output/training_plot_contrastive_{}.png'.format(args.prop))
+    plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
 
     #load and evaluate best model
     load_model(model, args.save_path)