REVA-QCAV/wandb.yaml
Laurent Fainsin 50d18a5b39 feat: random code I don't want to make commit messages for
Former-commit-id: b94db28e25c4ada7f69d65185198a701bb5d6bfd [formerly 2476ee5d84287e40c8fb341f569249dc8aaff3e5]
Former-commit-id: 0a4b7a1f925165172b009f8812d3083e70f10201
2022-07-12 11:18:03 +02:00

46 lines
651 B
YAML

DIR_TRAIN_IMG:
value: "/home/lilian/data_disk/lfainsin/train/"
DIR_VALID_IMG:
value: "/home/lilian/data_disk/lfainsin/test_batched_fast/"
DIR_SPHERE:
value: "/home/lilian/data_disk/lfainsin/spheres+real/"
FEATURES:
value: [8, 16, 32, 64]
N_CHANNELS:
value: 3
N_CLASSES:
value: 1
AMP:
value: True
PIN_MEMORY:
value: True
BENCHMARK:
value: True
DEVICE:
value: gpu
WORKERS:
value: 8
IMG_SIZE:
value: 512
SPHERES:
value: 3
EPOCHS:
value: 20
TRAIN_BATCH_SIZE:
value: 64 # 100
VAL_BATCH_SIZE:
value: 8 # 10
PREFETCH_FACTOR:
value: 16
LEARNING_RATE:
value: 1.0e-4
WEIGHT_DECAY:
value: 1.0e-8
MOMENTUM:
value: 0.9