picodet_l_416_coco.yml 947 字节
Newer Older
G
Guanghua Yu 已提交
1 2 3
_BASE_: [
  '../../datasets/coco_detection.yml',
  '../../runtime.yml',
G
Guanghua Yu 已提交
4 5 6
  '_base_/picodet_esnet.yml',
  '_base_/optimizer_300e.yml',
  '_base_/picodet_416_reader.yml',
G
Guanghua Yu 已提交
7 8
]

G
Guanghua Yu 已提交
9 10
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ESNet_x1_25_pretrained.pdparams
weights: output/picodet_l_416_coco/model_final
G
Guanghua Yu 已提交
11 12 13 14
find_unused_parameters: True
use_ema: true
cycle_epoch: 40
snapshot_epoch: 10
G
Guanghua Yu 已提交
15
epoch: 250
G
Guanghua Yu 已提交
16

G
Guanghua Yu 已提交
17 18 19 20 21
ESNet:
  scale: 1.25
  feature_maps: [4, 11, 14]
  act: hard_swish
  channel_ratio: [0.875, 0.5, 1.0, 0.625, 0.5, 0.75, 0.625, 0.625, 0.5, 0.625, 1.0, 0.625, 0.75]
G
Guanghua Yu 已提交
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37

CSPPAN:
  out_channels: 160

PicoHead:
  conv_feat:
    name: PicoFeat
    feat_in: 160
    feat_out: 160
    num_convs: 4
    num_fpn_stride: 4
    norm_type: bn
    share_cls_reg: True
  feat_in_chan: 160

TrainReader:
G
Guanghua Yu 已提交
38
  batch_size: 48
G
Guanghua Yu 已提交
39 40

LearningRate:
G
Guanghua Yu 已提交
41
  base_lr: 0.3
G
Guanghua Yu 已提交
42 43 44 45 46 47
  schedulers:
  - !CosineDecay
    max_epochs: 300
  - !LinearWarmup
    start_factor: 0.1
    steps: 300