retinanet_r50_fpn_90k_coco.py 639 B

123456789101112131415161718192021222324
  1. _base_ = 'retinanet_r50_fpn_1x_coco.py'
  2. # training schedule for 90k
  3. train_cfg = dict(
  4. _delete_=True,
  5. type='IterBasedTrainLoop',
  6. max_iters=90000,
  7. val_interval=10000)
  8. # learning rate policy
  9. param_scheduler = [
  10. dict(
  11. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  12. dict(
  13. type='MultiStepLR',
  14. begin=0,
  15. end=90000,
  16. by_epoch=False,
  17. milestones=[60000, 80000],
  18. gamma=0.1)
  19. ]
  20. train_dataloader = dict(sampler=dict(type='InfiniteSampler'))
  21. default_hooks = dict(checkpoint=dict(by_epoch=False, interval=10000))
  22. log_processor = dict(by_epoch=False)