Spaces:
tuan2308
/
Running on Zero

IDM-VTON / configs /HRNet /densepose_rcnn_HRFPN_HRNet_w48_s1x.yaml
IDM-VTON
update IDM-VTON Demo
938e515
raw
history blame
545 Bytes
_BASE_: "../Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "https://1drv.ms/u/s!Aus8VCZ_C_33dKvqI6pBZlifgJk"
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: [48, 96]
STAGE3:
NUM_CHANNELS: [48, 96, 192]
STAGE4:
NUM_CHANNELS: [48, 96, 192, 384]
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
CLIP_GRADIENTS:
ENABLED: True
CLIP_TYPE: "norm"
BASE_LR: 0.03