Spaces:
Runtime error
Runtime error
| 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 | |