From c9fbf050fb28e06cc66e12713a852e90bf6fa79b Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Thu, 24 Apr 2025 17:24:17 +0200
Subject: [PATCH] add : parameter config wandb log
---
image_ref/main_sweep.py | 5 +++--
image_ref/sweep_train.py | 2 +-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/image_ref/main_sweep.py b/image_ref/main_sweep.py
index f7c4730..ac6d761 100644
--- a/image_ref/main_sweep.py
+++ b/image_ref/main_sweep.py
@@ -21,7 +21,7 @@ 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"]},
},
- "max_iter": 10
+ "max_iter": 100
}
for i in range(sweep_configuration["max_iter"]):
@@ -37,5 +37,6 @@ if __name__ == '__main__':
run_config[p] = random.uniform(v["min"],v["max"])
elif v["distribution"]=="log_uniform_values":
run_config[p] = np.exp(random.uniform(np.log(v["min"]), np.log(v["max"])))
- print('Launching run')
+ print('Launching run ',i)
+ print('Parameter : ', sweep_configuration["parameters"])
run_duo(run_config)
diff --git a/image_ref/sweep_train.py b/image_ref/sweep_train.py
index d2a273a..418256a 100644
--- a/image_ref/sweep_train.py
+++ b/image_ref/sweep_train.py
@@ -90,7 +90,7 @@ def run_duo(args):
os.environ["WANDB_MODE"] = "offline"
os.environ["WANDB_DIR"] = os.path.abspath("./wandb_run")
- wdb.init(project="param_sweep_contrastive", dir='./wandb_run')
+ wdb.init(project="param_sweep_contrastive", dir='./wandb_run', config=args)
print('Wandb initialised')
# load data
--
GitLab