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

G
Guanghua Yu 已提交
9
weights: output/picodet_m_416_coco/best_model
G
Guanghua Yu 已提交
10 11
find_unused_parameters: True
use_ema: true
G
Guanghua Yu 已提交
12
epoch: 250
G
Guanghua Yu 已提交
13
snapshot_epoch: 10
G
Guanghua Yu 已提交
14 15 16 17 18 19 20 21 22 23 24 25

TrainReader:
  batch_size: 48

LearningRate:
  base_lr: 0.24
  schedulers:
  - !CosineDecay
    max_epochs: 300
  - !LinearWarmup
    start_factor: 0.1
    steps: 300