From 86417c93b03d4fa88bdbd7f4e0a9494d9e06a90a Mon Sep 17 00:00:00 2001 From: Schneider Leo <leo.schneider@etu.ec-lyon.fr> Date: Tue, 20 May 2025 15:19:09 +0200 Subject: [PATCH] fix : wandb log barlow --- barlow_twin_like/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py index 01effe1..353a015 100644 --- a/barlow_twin_like/main.py +++ b/barlow_twin_like/main.py @@ -181,7 +181,8 @@ def run(): 'hparam/lr': args.lr, 'hparam/data_train': args.dataset_train_dir, 'hparam/data_val': args.dataset_train_dir, 'hparam/model': args.model, - 'hparam/ref_dir': args.dataset_ref_dir}) + 'hparam/ref_dir': args.dataset_ref_dir, + 'hparam/batch': args.batch_size}) print('Wandb initialised') # load data -- GitLab