fix : path

image_ref/main.py

@@ -103,7 +103,7 @@ def run_duo(args):
             val_acc.append(acc)
             if loss < best_loss :
                 save_model(model,args.save_path)
-                loss = acc
+                best_loss = loss
     # plot and save training figs
     plt.clf()
     plt.subplot(2, 1, 1)