picodet_m_416_coco_lcnet.yml 449 B

12345678910111213141516171819202122232425
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '_base_/picodet_v2.yml',
  5. '_base_/optimizer_300e.yml',
  6. '_base_/picodet_416_reader.yml',
  7. ]
  8. weights: output/picodet_m_416_coco/best_model
  9. find_unused_parameters: True
  10. use_ema: true
  11. epoch: 250
  12. snapshot_epoch: 10
  13. TrainReader:
  14. batch_size: 48
  15. LearningRate:
  16. base_lr: 0.24
  17. schedulers:
  18. - !CosineDecay
  19. max_epochs: 300
  20. - !LinearWarmup
  21. start_factor: 0.1
  22. steps: 300