retinanet_r50_fpn_1x_coco.py 312 B

12345678910
  1. _base_ = [
  2. '../_base_/models/retinanet_r50_fpn.py',
  3. '../_base_/datasets/coco_detection.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py',
  5. './retinanet_tta.py'
  6. ]
  7. # optimizer
  8. optim_wrapper = dict(
  9. optimizer=dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001))