diff --git a/src/train.py b/src/train.py index d111e38..a1e54c8 100644 --- a/src/train.py +++ b/src/train.py @@ -12,7 +12,7 @@ CONFIG = { "DIR_TRAIN_IMG": "/home/lilian/data_disk/lfainsin/train/", "DIR_VALID_IMG": "/home/lilian/data_disk/lfainsin/val/", "DIR_TEST_IMG": "/home/lilian/data_disk/lfainsin/test/", - "DIR_SPHERE": "/home/lilian/data_disk/lfainsin/spheres/", + "DIR_SPHERE": "/home/lilian/data_disk/lfainsin/realspheres/", "FEATURES": [8, 16, 32, 64], "N_CHANNELS": 3, "N_CLASSES": 1,