Skip to content

GitLab

"tools/cfgs/kitti_models/pointpillar_newaugs.yaml" did not exist on "71a45080af988eaa796b850f685fbfd41976a2c4"
Switch branch/tag
Find file Normal viewHistoryPermalink
internimage_l_22kto1k_384.yaml 648 Bytes
Newer Older
PRC-Huang's avatar
release classification  
PRC-Huang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37 
DATA:
  IMG_SIZE: 384
  IMG_ON_MEMORY: True
AUG:
  MIXUP: 0.0
  CUTMIX: 0.0
  REPROB: 0.0
MODEL:
  TYPE: intern_image
  DROP_PATH_RATE: 0.1
  LABEL_SMOOTHING: 0.3
  INTERN_IMAGE:
    CORE_OP: 'DCNv3'
    DEPTHS: [5, 5, 22, 5]
    GROUPS: [10, 20, 40, 80]
    CHANNELS: 160
    LAYER_SCALE: 1e-5
    OFFSET_SCALE: 2.0
    MLP_RATIO: 4.0
    POST_NORM: True
TRAIN:
  EMA:
    ENABLE: true
    DECAY: 0.9999
  EPOCHS: 20
  WARMUP_EPOCHS: 2
  WEIGHT_DECAY: 0.05
  BASE_LR: 2e-05 # 512
  WARMUP_LR: .0
  MIN_LR: .0
  LR_LAYER_DECAY: true
  LR_LAYER_DECAY_RATIO: 0.9
  USE_CHECKPOINT: true
  OPTIMIZER:
    DCN_LR_MUL: 0.1
AMP_OPT_LEVEL: O0
EVAL_FREQ: 1