REVA-QCAV/wandb.yaml
Laurent Fainsin c312513eff feat: checkpointing + torchmetrics + ...
Former-commit-id: 781a11f646964567b3bc6831caa5e380748b84e1 [formerly 7ea5f37ecb7fd5c95e5d1b7b1aa899d986ecff2a]
Former-commit-id: 06974fa163a6a8eb881d5981f9c6debe63f1b4bb
2022-09-05 16:04:52 +02:00

45 lines
614 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: 50
TRAIN_BATCH_SIZE:
value: 10
VALID_BATCH_SIZE:
value: 2
PREFETCH_FACTOR:
value: 2
LEARNING_RATE:
value: 0.0005
WEIGHT_DECAY:
value: 0.0005
MOMENTUM:
value: 0.9