From 9fe9c3415594ee2b811d7dc84402523c74e22640 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 25 Mar 2025 10:49:34 +0100
Subject: [PATCH] small fix main

---
 main.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 4be686c..30f7843 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
-- 
GitLab