paa_r50_fpn_2x_coco.py 399 B

123456789101112131415161718
  1. _base_ = './paa_r50_fpn_1x_coco.py'
  2. max_epochs = 24
  3. # learning rate
  4. param_scheduler = [
  5. dict(
  6. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  7. dict(
  8. type='MultiStepLR',
  9. begin=0,
  10. end=max_epochs,
  11. by_epoch=True,
  12. milestones=[16, 22],
  13. gamma=0.1)
  14. ]
  15. # training schedule for 2x
  16. train_cfg = dict(max_epochs=max_epochs)