REVA-QCAV/wandb.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