From e36e38d7811eb51948e61c3ac26f9800acaad49c Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 31 Mar 2025 15:11:27 +0200
Subject: [PATCH] fix : file config
---
image_ref/config.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/image_ref/config.py b/image_ref/config.py
index 455bfa13..58a4c48b 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')
--
GitLab