projet-long/wandb.yaml
Laurent Fainsin 8d6115afb2 feat: add more config fields
Former-commit-id: 4b288d6c47aec460704348a003b48620327c218d [formerly 1af19e81906cd94483f7e9aadc7fdd72d35dc5e1]
Former-commit-id: 605c529e3db6dde13cbbcf41f9517c3380b69b29
2022-08-26 11:10:49 +02:00

52 lines
677 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/"
FEATURES:
value: [8, 16, 32, 64]
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
IMG_SIZE:
value: 512
SPHERES:
value: 3
EPOCHS:
value: 10
TRAIN_BATCH_SIZE:
value: 8
VAL_BATCH_SIZE:
value: 0
PREFETCH_FACTOR:
value: 2
LEARNING_RATE:
value: 0.005
WEIGHT_DECAY:
value: 0.0005
MOMENTUM:
value: 0.9