REVA-QCAV/wandb.yaml
Laurent Fainsin 4dab157dda feat: WIP, replacing U-Net by Mask R-CNN
Former-commit-id: f51a572adac901ff588e3a467f39ecd26376e617 [formerly 376595d7e5f906928379e25c1246e304b96b156d]
Former-commit-id: 3f4772ba3483702be6e5f7a29f06be93eb1f3bb2
2022-08-24 14:56:41 +02:00

46 lines
615 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
DEVICE:
value: gpu
WORKERS:
value: 8
IMG_SIZE:
value: 512
SPHERES:
value: 3
EPOCHS:
value: 3
TRAIN_BATCH_SIZE:
value: 128 # 100
VAL_BATCH_SIZE:
value: 8 # 10
PREFETCH_FACTOR:
value: 2
LEARNING_RATE:
value: 1.0e-4
WEIGHT_DECAY:
value: 1.0e-8
MOMENTUM:
value: 0.9