diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py
index 01effe173511771c7f8650c1583e1892be2cd407..353a01593a46f0aec305f4d3cb72bef86487569f 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