tridentnet_r50-caffe_ms-3x_coco.py 431 B

123456789101112131415161718
  1. _base_ = 'tridentnet_r50-caffe_ms-1x_coco.py'
  2. # learning rate
  3. max_epochs = 36
  4. train_cfg = dict(
  5. type='EpochBasedTrainLoop', max_epochs=max_epochs, val_interval=1)
  6. param_scheduler = [
  7. dict(
  8. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  9. dict(
  10. type='MultiStepLR',
  11. begin=0,
  12. end=max_epochs,
  13. by_epoch=True,
  14. milestones=[28, 34],
  15. gamma=0.1)
  16. ]