search

salesforce / ALBEF

Code for ALBEF: a new vision-language pre-training method
BSD 3-Clause "New" or "Revised" License
1.45k stars 193 forks source link

state_dict = checkpoint['model'] KeyError: 'model' When I using flickr30k.pth #120

Open MARS-J opened 1 year ago

MARS-J commented 1 year ago

I am using the Windows platform， The command I executed was：“python Retrieval.py --config ./configs/Retrieval_flickr.yaml --output_dir output/Retrieval_flickr --checkpoint pretrain\flickr30k.pth”， The error is： “Traceback (most recent call last): File "Retrieval.py", line 376, in main(args, config) File "Retrieval.py", line 257, in main state_dict = checkpoint['model'] KeyError: 'model' ” I would appreciate it very much if someone could help me solve this problem.

T0305 commented 10 months ago

Did you solve this problem? I also encountered this problem, can you tell me how to solve it? Thank you so much

shyammarjit commented 4 months ago

I am also getting the same error.