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-b5_fpn_1x_coco.py 515 B

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

No Description

Contributors (3)