faster-rcnn_r50-scratch_fpn_gn-all_6x_coco.py 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. _base_ = [
  2. '../_base_/models/faster-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_detection.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. optim_wrapper = dict(paramwise_cfg=dict(norm_decay_mult=0.))
  20. max_epochs = 73
  21. param_scheduler = [
  22. dict(
  23. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  24. dict(
  25. type='MultiStepLR',
  26. begin=0,
  27. end=max_epochs,
  28. by_epoch=True,
  29. milestones=[65, 71],
  30. gamma=0.1)
  31. ]
  32. train_cfg = dict(max_epochs=max_epochs)
  33. # only keep latest 3 checkpoints
  34. default_hooks = dict(checkpoint=dict(max_keep_ckpts=3))