wdb : batch_size

main.py

@@ -238,7 +238,7 @@ def run_duo(args):
 
         wdb.init(project="experiment_setting", dir='./wandb_run', config={
                  'hparam/lr':args.lr,'hparam/data_train':args.dataset_train_dir,'hparam/data_val':args.dataset_train_dir,
-                 'hparam/model_type':args.model_type, 'hparam/setting':args.setting})
+                 'hparam/model_type':args.model_type, 'hparam/setting':args.setting,'hparam/batch_size':args.batch_size })
 
         print('Wandb initialised')