forked from PulseFocusPlatform/PulseFocusPlatform
43 lines
775 B
YAML
43 lines
775 B
YAML
|
_BASE_: [
|
||
|
|
'../datasets/voc.yml',
|
||
|
|
'../runtime.yml',
|
||
|
|
'./_base_/ppyolov2_r50vd_dcn.yml',
|
||
|
|
'./_base_/optimizer_365e.yml',
|
||
|
|
'./_base_/ppyolov2_reader.yml',
|
||
|
|
]
|
||
|
|
|
||
|
|
snapshot_epoch: 83
|
||
|
|
weights: output/ppyolov2_r50vd_dcn_voc/model_final
|
||
|
|
|
||
|
|
TrainReader:
|
||
|
|
mixup_epoch: 350
|
||
|
|
batch_size: 12
|
||
|
|
|
||
|
|
# set collate_batch to false because ground-truth info is needed
|
||
|
|
# on voc dataset and should not collate data in batch when batch size
|
||
|
|
# is larger than 1.
|
||
|
|
EvalReader:
|
||
|
|
collate_batch: false
|
||
|
|
|
||
|
|
epoch: 583
|
||
|
|
|
||
|
|
LearningRate:
|
||
|
|
base_lr: 0.00333
|
||
|
|
schedulers:
|
||
|
|
- !PiecewiseDecay
|
||
|
|
gamma: 0.1
|
||
|
|
milestones:
|
||
|
|
- 466
|
||
|
|
- 516
|
||
|
|
- !LinearWarmup
|
||
|
|
start_factor: 0.
|
||
|
|
steps: 4000
|
||
|
|
|
||
|
|
OptimizerBuilder:
|
||
|
|
optimizer:
|
||
|
|
momentum: 0.9
|
||
|
|
type: Momentum
|
||
|
|
regularizer:
|
||
|
|
factor: 0.0005
|
||
|
|
type: L2
|