REVA-QCAV/wandb.yaml

45 lines
611 B
YAML

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