diff --git a/image_ref/config.py b/image_ref/config.py
index 455bfa13fbd44b0552ecbd76debae6e6a428a87e..58a4c48bd9c1f0078ca99c06194db9c031dd9db4 100644
--- a/image_ref/config.py
+++ b/image_ref/config.py
@@ -10,6 +10,7 @@ def load_args_contrastive():
     parser.add_argument('--noise_threshold', type=int, default=0)
     parser.add_argument('--lr', type=float, default=0.001)
     parser.add_argument('--batch_size', type=int, default=64)
+    parser.add_argument('--positive_prop', type=int, default=None)
     parser.add_argument('--model', type=str, default='ResNet18')
     parser.add_argument('--model_type', type=str, default='duo')
     parser.add_argument('--dataset_dir', type=str, default='data/processed_data/npy_image/data_training')