24 lines
467 B
YAML
24 lines
467 B
YAML
_BASE_: [
|
|
'../datasets/coco_detection.yml',
|
|
'./_base_/faster_rcnn_hrnetv2p_w18.yml',
|
|
'../faster_rcnn/_base_/optimizer_1x.yml',
|
|
'../faster_rcnn/_base_/faster_fpn_reader.yml',
|
|
'../runtime.yml',
|
|
]
|
|
|
|
weights: output/faster_rcnn_hrnetv2p_w18_2x_coco/model_final
|
|
epoch: 24
|
|
|
|
LearningRate:
|
|
base_lr: 0.02
|
|
schedulers:
|
|
- !PiecewiseDecay
|
|
gamma: 0.1
|
|
milestones: [16, 22]
|
|
- !LinearWarmup
|
|
start_factor: 0.1
|
|
steps: 1000
|
|
|
|
TrainReader:
|
|
batch_size: 2
|