retinanet_r101_fpn_ms-640-800-3x_coco.py 343 B

123456789
  1. _base_ = ['../_base_/models/retinanet_r50_fpn.py', '../common/ms_3x_coco.py']
  2. # optimizer
  3. model = dict(
  4. backbone=dict(
  5. depth=101,
  6. init_cfg=dict(type='Pretrained',
  7. checkpoint='torchvision://resnet101')))
  8. optim_wrapper = dict(
  9. optimizer=dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001))