REVA-QCAV/wandb.yaml

DIR_TRAIN_IMG:
  value: "/media/disk1/lfainsin/TRAIN_prerender/"
DIR_VALID_IMG:
  value: "/media/disk1/lfainsin/TEST_tmp_mrcnn/"
# 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: 32
SEED:
  value: 69420
DEVICE:
  value: gpu
WORKERS:
  value: 16

EPOCHS:
  value: 50
TRAIN_BATCH_SIZE:
  value: 6
VALID_BATCH_SIZE:
  value: 2
PREFETCH_FACTOR:
  value: 2

LEARNING_RATE:
  value: 0.001
WEIGHT_DECAY:
  value: 0.0001
MOMENTUM:
  value: 0.9