small fix main

main.py

@@ -257,7 +257,6 @@ def load_model(model, path):
 if __name__ == '__main__':
     args = load_args()
     if args.model_type=='duo':
-        # run_duo(args)
-        data_train, data_test = load_data_duo(base_dir=args.dataset_dir, batch_size=args.batch_size)
+        run_duo(args)
     else :
         run(args)
\ No newline at end of file