fix : typo

image_ref/main.py

@@ -169,11 +169,11 @@ def run_duo(args):
         plt.tight_layout()
 
         plt.show()
-        plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
+        plt.savefig(args.base_out+'_training_plot.png')
 
     # load and evaluate best model
     load_model(model, args.save_path)
-    if args.args.dataset_test_dir is not None :
+    if args.dataset_test_dir is not None :
         make_prediction_duo(model, data_test_batch,args.base_out+'_confusion_matrix_test.png',
                             args.base_out+'confidence_matrix_.png')