Update retinanet_swin-t-p4-w7_fpn_1x_coco.py (#6973)
Fix the error encountered during loading the pretrained backbone as follows, 2022-01-10 09:45:12,308 - mmdet - INFO - load checkpoint from http path: https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_small_patch4_window7_224.pth Traceback (most recent call last): File "tools/train.py", line 185, in <module> main() File "tools/train.py", line 159, in main model.init_weights() File "/home/shengliliu/anaconda3/envs/cots/lib/python3.7/site-packages/mmcv/runner/base_module.py", line 116, in init_weights m.init_weights() File "/home/shengliliu/anaconda3/envs/cots/lib/python3.7/site-packages/mmdet/models/backbones/swin.py", line 706, in init_weights if list(state_dict.keys())[0].startswith('module.'): IndexError: list index out of rangepull/6998/head^2
parent
3a00ade9c3
commit
12d0d6f2ca
1 changed files with 1 additions and 0 deletions
Loading…
Reference in new issue