REVA-QCAV/wandb.yaml
Laurent Fainsin 562ef110af feat: basic validation
Former-commit-id: 9d6ac1da57e613d3fa4db7ba68560509611c2b79 [formerly f588a5a0781e7375baa6ce5a4723a7b0d557316a]
Former-commit-id: 7415ee71e570da13f9de361131156ba0d2131d8a
2022-08-30 10:20:32 +02:00

45 lines
601 B
YAML

DIR_TRAIN_IMG:
value: "/media/disk1/lfainsin/BACKGROUND/"
DIR_VALID_IMG:
value: "/media/disk1/lfainsin/TEST_batched/"
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: 16
SEED:
value: 69420
DEVICE:
value: gpu
WORKERS:
value: 16
EPOCHS:
value: 10
TRAIN_BATCH_SIZE:
value: 10
VALID_BATCH_SIZE:
value: 2
PREFETCH_FACTOR:
value: 2
LEARNING_RATE:
value: 0.005
WEIGHT_DECAY:
value: 0.0005
MOMENTUM:
value: 0.9