DIR_TRAIN_IMG: value: "/media/disk1/lfainsin/BACKGROUND/" DIR_VALID_IMG: value: "/media/disk1/lfainsin/TEST_batched/" DIR_SPHERE: value: "/media/disk1/lfainsin/SPHERES/" FEATURES: value: [8, 16, 32, 64] N_CHANNELS: value: 3 N_CLASSES: value: 1 AMP: value: True PIN_MEMORY: value: True BENCHMARK: value: True DEVICE: value: gpu WORKERS: value: 8 IMG_SIZE: value: 512 SPHERES: value: 3 EPOCHS: value: 3 TRAIN_BATCH_SIZE: value: 128 # 100 VAL_BATCH_SIZE: value: 8 # 10 PREFETCH_FACTOR: value: 2 LEARNING_RATE: value: 1.0e-4 WEIGHT_DECAY: value: 1.0e-8 MOMENTUM: value: 0.9