mask-rcnn_r50-scratch_fpn_gn-all_6x_coco.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. _base_ = [
  2. '../_base_/models/mask-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_instance.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. norm_cfg = dict(type='GN', num_groups=32, requires_grad=True)
  7. model = dict(
  8. backbone=dict(
  9. frozen_stages=-1,
  10. zero_init_residual=False,
  11. norm_cfg=norm_cfg,
  12. init_cfg=None),
  13. neck=dict(norm_cfg=norm_cfg),
  14. roi_head=dict(
  15. bbox_head=dict(
  16. type='Shared4Conv1FCBBoxHead',
  17. conv_out_channels=256,
  18. norm_cfg=norm_cfg),
  19. mask_head=dict(norm_cfg=norm_cfg)))
  20. optim_wrapper = dict(paramwise_cfg=dict(norm_decay_mult=0.))
  21. max_epochs = 73
  22. param_scheduler = [
  23. dict(
  24. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  25. dict(
  26. type='MultiStepLR',
  27. begin=0,
  28. end=max_epochs,
  29. by_epoch=True,
  30. milestones=[65, 71],
  31. gamma=0.1)
  32. ]
  33. train_cfg = dict(max_epochs=max_epochs)
  34. # only keep latest 3 checkpoints
  35. default_hooks = dict(checkpoint=dict(max_keep_ckpts=3))