DIR_TRAIN_IMG: value: "/media/disk1/lfainsin/TRAIN_prerender/" DIR_VALID_IMG: value: "/media/disk1/lfainsin/TEST_tmp_mrcnn/" # DIR_SPHERE: # value: "/media/disk1/lfainsin/SPHERES/" N_CHANNELS: value: 3 N_CLASSES: value: 1 AMP: value: True PIN_MEMORY: value: True BENCHMARK: value: True DETERMINISTIC: value: False PRECISION: value: 32 SEED: value: 69420 DEVICE: value: gpu WORKERS: value: 16 EPOCHS: value: 50 TRAIN_BATCH_SIZE: value: 6 VALID_BATCH_SIZE: value: 2 PREFETCH_FACTOR: value: 2 LEARNING_RATE: value: 0.001 WEIGHT_DECAY: value: 0.0001 MOMENTUM: value: 0.9