fix : config model

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"},