Spaces:
Saad0KH
/
Running on Zero

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