retinanet_pvt-l_fpn_1x_coco.py 349 B

12345678
  1. _base_ = 'retinanet_pvt-t_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. num_layers=[3, 8, 27, 3],
  5. init_cfg=dict(checkpoint='https://github.com/whai362/PVT/'
  6. 'releases/download/v2/pvt_large.pth')))
  7. # Enable automatic-mixed-precision training with AmpOptimWrapper.
  8. optim_wrapper = dict(type='AmpOptimWrapper')