ImportError: cannot import name 'wrap_fp16_model'

[xiaosa96]

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