diff --git a/main.py b/main.py
index 4be686c5c7efb36825fc341afcf2c4d2ab0c4e87..30f7843b0f6285ad5c47bf5d31e0176e2666e70e 100644
--- a/main.py
+++ b/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