ImportError: cannot import name 'wrap_fp16_model'

implus / GFocalV2

Generalized Focal Loss V2: Learning Reliable Localization Quality Estimation for Dense Object Detection, CVPR2021

Apache License 2.0

475 stars 54 forks source link

ImportError: cannot import name 'wrap_fp16_model' #15

Closed xiaosa96 closed 3 years ago

xiaosa96 commented 3 years ago

hello，When I use the command bash./tools/dist_test.sh configs/gfocal/gfocal_r50_fpn_ms2x.py work_dirs/gfocal_r50_fpn_ms2x/latest.pth 4 or running the test.py file with vscode. There are a ImporError: Cannot import name 'WRAP_FP16_MODEL'. I would like to ask you how to solve this problem?My Torch environment is 1.5.0, TorchVision is 0.6.0, and MMCV_Full version is 1.0.5