From a33f56395e613829594d7e9189d6833ced7c0fe8 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 23 Apr 2025 15:12:14 +0200
Subject: [PATCH] chang : wdb hparam exported as config
---
image_ref/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index bd93c6b..20a099f 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -101,10 +101,10 @@ def run_duo(args):
os.environ["WANDB_MODE"] = "offline"
os.environ["WANDB_DIR"] = os.path.abspath("./wandb_run")
- wdb.init(project="contrastive_classification", dir='./wandb_run')
- wdb.log({'hparam/optimizer':args.opti,'hparam/p_prop':args.positive_prop,'hparam/sampler':args.sampler,
+ wdb.init(project="contrastive_classification", dir='./wandb_run', config={'hparam/optimizer':args.opti,'hparam/p_prop':args.positive_prop,'hparam/sampler':args.sampler,
'hparam/lr':args.lr,'hparam/data_train':args.dataset_train_dir,'hparam/data_val':args.dataset_train_dir,
'hparam/noise':args.noise_threshold,'hparam/model':args.model, 'hparam/ref_dir':args.dataset_ref_dir})
+
print('Wandb initialised')
# load data
data_train, data_val_batch, data_test_batch = load_data_duo(base_dir_train=args.dataset_train_dir,
--
GitLab