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