diff --git a/image_ref/main.py b/image_ref/main.py
index 933794ad4ebbe513cac01d1246afd32868d794cd..bd93c6b47a872abd6b158e10bf694b65304551df 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -104,7 +104,7 @@ def run_duo(args):
         wdb.init(project="contrastive_classification", dir='./wandb_run')
         wdb.log({'hparam/optimizer':args.opti,'hparam/p_prop':args.positive_prop,'hparam/sampler':args.sampler,
                  'hparam/lr':args.lr,'hparam/data_train':args.dataset_train_dir,'hparam/data_val':args.dataset_train_dir,
-                 'hparam/noise':args.noise_threshold,'hparam/model':args.model})
+                 'hparam/noise':args.noise_threshold,'hparam/model':args.model, 'hparam/ref_dir':args.dataset_ref_dir})
         print('Wandb initialised')
     # load data
     data_train, data_val_batch, data_test_batch = load_data_duo(base_dir_train=args.dataset_train_dir,