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