mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-08 14:39:00 +00:00
50d18a5b39
Former-commit-id: b94db28e25c4ada7f69d65185198a701bb5d6bfd [formerly 2476ee5d84287e40c8fb341f569249dc8aaff3e5] Former-commit-id: 0a4b7a1f925165172b009f8812d3083e70f10201
46 lines
651 B
YAML
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
|