From da85a5bbe4bf9ac1e1b9eb86b2b25288f5d79402 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Thu, 24 Apr 2025 15:13:41 +0200
Subject: [PATCH] fix : config model
---
image_ref/main_sweep.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/image_ref/main_sweep.py b/image_ref/main_sweep.py
index 96d46f2..f7c4730 100644
--- a/image_ref/main_sweep.py
+++ b/image_ref/main_sweep.py
@@ -15,7 +15,7 @@ if __name__ == '__main__':
"batch_size": {"value": 64},
"positive_prop": {"distribution" : "uniform","max": 95., "min": 5.},
"opti": {"value": "adam"},
- "model": {"value": "resnet18"},
+ "model": {"value": "ResNet18"},
"sampler": {"values": ["random","balanced"]},
"dataset_train_dir": {"value": "data/processed_data_wiff/npy_image/train_data"},
"dataset_val_dir": {"value": "data/processed_data_wiff/npy_image/test_data"},
--
GitLab