2022-07-11 13:34:05 +00:00
|
|
|
DIR_TRAIN_IMG:
|
2022-08-23 13:27:23 +00:00
|
|
|
value: "/media/disk1/lfainsin/BACKGROUND/"
|
2022-07-11 13:34:05 +00:00
|
|
|
DIR_VALID_IMG:
|
2022-08-23 13:27:23 +00:00
|
|
|
value: "/media/disk1/lfainsin/TEST_batched/"
|
2022-07-11 13:34:05 +00:00
|
|
|
DIR_SPHERE:
|
2022-08-23 13:27:23 +00:00
|
|
|
value: "/media/disk1/lfainsin/SPHERES/"
|
2022-07-11 13:34:05 +00:00
|
|
|
|
|
|
|
N_CHANNELS:
|
2022-07-11 14:13:16 +00:00
|
|
|
value: 3
|
2022-07-11 13:34:05 +00:00
|
|
|
N_CLASSES:
|
2022-07-11 14:13:16 +00:00
|
|
|
value: 1
|
2022-07-11 13:34:05 +00:00
|
|
|
|
|
|
|
AMP:
|
|
|
|
value: True
|
|
|
|
PIN_MEMORY:
|
|
|
|
value: True
|
|
|
|
BENCHMARK:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: True
|
|
|
|
DETERMINISTIC:
|
2022-08-25 08:40:50 +00:00
|
|
|
value: False
|
2022-08-26 09:10:49 +00:00
|
|
|
PRECISION:
|
|
|
|
value: 16
|
|
|
|
SEED:
|
|
|
|
value: 69420
|
2022-07-11 13:34:05 +00:00
|
|
|
DEVICE:
|
|
|
|
value: gpu
|
|
|
|
WORKERS:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: 16
|
2022-07-11 13:34:05 +00:00
|
|
|
|
|
|
|
EPOCHS:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: 10
|
2022-07-11 14:13:16 +00:00
|
|
|
TRAIN_BATCH_SIZE:
|
2022-08-30 08:20:32 +00:00
|
|
|
value: 10
|
|
|
|
VALID_BATCH_SIZE:
|
|
|
|
value: 2
|
2022-07-12 09:18:03 +00:00
|
|
|
PREFETCH_FACTOR:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: 2
|
2022-07-11 13:34:05 +00:00
|
|
|
|
|
|
|
LEARNING_RATE:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: 0.005
|
2022-07-11 13:34:05 +00:00
|
|
|
WEIGHT_DECAY:
|
2022-08-26 09:10:49 +00:00
|
|
|
value: 0.0005
|
2022-07-11 13:34:05 +00:00
|
|
|
MOMENTUM:
|
|
|
|
value: 0.9
|