File size: 545 Bytes
938e515 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
_BASE_: "../Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "https://1drv.ms/u/s!Aus8VCZ_C_33ck0gvo5jfoWBOPo"
BACKBONE:
NAME: "build_hrfpn_backbone"
RPN:
IN_FEATURES: ['p1', 'p2', 'p3', 'p4', 'p5']
ROI_HEADS:
IN_FEATURES: ['p1', 'p2', 'p3', 'p4', 'p5']
HRNET:
STAGE2:
NUM_CHANNELS: [40, 80]
STAGE3:
NUM_CHANNELS: [40, 80, 160]
STAGE4:
NUM_CHANNELS: [40, 80, 160, 320]
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
CLIP_GRADIENTS:
ENABLED: True
CLIP_TYPE: "norm"
BASE_LR: 0.03
|