vfnet_r101_fpn_2x_coco.py 519 B

1234567891011121314151617181920
  1. _base_ = './vfnet_r50_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. depth=101,
  5. init_cfg=dict(type='Pretrained',
  6. checkpoint='torchvision://resnet101')))
  7. # learning policy
  8. max_epochs = 24
  9. param_scheduler = [
  10. dict(type='LinearLR', start_factor=0.1, by_epoch=False, begin=0, end=500),
  11. dict(
  12. type='MultiStepLR',
  13. begin=0,
  14. end=max_epochs,
  15. by_epoch=True,
  16. milestones=[16, 22],
  17. gamma=0.1)
  18. ]
  19. train_cfg = dict(max_epochs=max_epochs)