ICON / lib /pymaf /configs /pymaf_config.yaml
Yuliang's picture
done
2d5f249
raw
history blame contribute delete
765 Bytes
SOLVER:
MAX_ITER: 500000
TYPE: Adam
BASE_LR: 0.00005
GAMMA: 0.1
STEPS: [0]
EPOCHS: [0]
DEBUG: False
LOGDIR: ''
DEVICE: cuda
NUM_WORKERS: 8
SEED_VALUE: -1
LOSS:
KP_2D_W: 300.0
KP_3D_W: 300.0
SHAPE_W: 0.06
POSE_W: 60.0
VERT_W: 0.0
INDEX_WEIGHTS: 2.0
# Loss weights for surface parts. (24 Parts)
PART_WEIGHTS: 0.3
# Loss weights for UV regression.
POINT_REGRESSION_WEIGHTS: 0.5
TRAIN:
NUM_WORKERS: 8
BATCH_SIZE: 64
PIN_MEMORY: True
TEST:
BATCH_SIZE: 32
MODEL:
PyMAF:
BACKBONE: 'res50'
MLP_DIM: [256, 128, 64, 5]
N_ITER: 3
AUX_SUPV_ON: True
DP_HEATMAP_SIZE: 56
RES_MODEL:
DECONV_WITH_BIAS: False
NUM_DECONV_LAYERS: 3
NUM_DECONV_FILTERS:
- 256
- 256
- 256
NUM_DECONV_KERNELS:
- 4
- 4
- 4