You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

retinanet_pvtv2-b4_fpn_1x_coco.py 482 B

2 years ago
12345678910111213
  1. _base_ = 'retinanet_pvtv2-b0_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. embed_dims=64,
  5. num_layers=[3, 8, 27, 3],
  6. init_cfg=dict(checkpoint='https://github.com/whai362/PVT/'
  7. 'releases/download/v2/pvt_v2_b4.pth')),
  8. neck=dict(in_channels=[64, 128, 320, 512]))
  9. # optimizer
  10. optimizer = dict(
  11. _delete_=True, type='AdamW', lr=0.0001 / 1.4, weight_decay=0.0001)
  12. # dataset settings
  13. data = dict(samples_per_gpu=1, workers_per_gpu=1)

No Description

Contributors (2)