forked from PulseFocusPlatform/PulseFocusPlatform
43 lines
767 B
YAML
43 lines
767 B
YAML
_BASE_: [
|
|
'../datasets/voc.yml',
|
|
'../runtime.yml',
|
|
'./_base_/ppyolo_r50vd_dcn.yml',
|
|
'./_base_/optimizer_1x.yml',
|
|
'./_base_/ppyolo_reader.yml',
|
|
]
|
|
|
|
snapshot_epoch: 83
|
|
weights: output/ppyolo_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
|