From ba0ac47cceb79a98ec0fdbb1c236c3690841d187 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Thu, 24 Apr 2025 14:20:36 +0200
Subject: [PATCH] add : wandb sweep debug print
---
image_ref/main_sweep.py | 16 ++++++++++------
1 file changed, 10 insertions(+), 6 deletions(-)
diff --git a/image_ref/main_sweep.py b/image_ref/main_sweep.py
index a9e1696..64fb7c7 100644
--- a/image_ref/main_sweep.py
+++ b/image_ref/main_sweep.py
@@ -1,10 +1,8 @@
+import time
import wandb as wdb
-from config import load_args_contrastive
-
if __name__ == '__main__':
- args = load_args_contrastive()
sweep_configuration = {
"program": "sweep_train.py",
"method": "random",
@@ -23,9 +21,15 @@ if __name__ == '__main__':
"dataset_val_dir": {"value": "data/processed_data_wiff/npy_image/test_data"},
"dataset_ref_dir": {"values": ["image_ref/img_ref","image_ref/img_ref_count_th_10","image_ref/img_ref_count_th_5"]},
},
+ "controller":{
+ "type": "local"},
}
- sweep_id = wdb.sweep(sweep=sweep_configuration, project="param_sweep_contrastive")
+ sweep_id = wdb.sweep(sweep_configuration)
+
+ # Start the local controller
sweep = wdb.controller(sweep_id)
- sweep.configure_controller(type="local")
- sweep.run(verbose=True,print_debug=True,print_status=True,print_actions=True)
+ while not sweep.done():
+ sweep.print_status()
+ sweep.step()
+ time.sleep(5)
--
GitLab