errors happend during the loading process of pretrained detr-r50-e632da11.pth

ZhanYang-nwpu / RSVG-pytorch

RSVG: Exploring Data and Model for Visual Grounding on Remote Sensing Data, 2022

106 stars 4 forks source link

errors happend during the loading process of pretrained detr-r50-e632da11.pth #13

Closed LANMNG closed 1 year ago

LANMNG commented 1 year ago

Traceback (most recent call last):
File "main.py", line 541, in
main()
File "main.py", line 170, in main
model = load_pretrain(model, args, logging) File "/data/users/lanmeng/projects/RSVG/utils/checkpoint.py", line 20, in load_pretrain
pretrained_dict = checkpoint['state_dict'] KeyError: 'state_dict'