MCG-NJU/VideoMAE

KeyError: 'model'

fengjingchehu opened this issue · 1 comments

VideoMAE-main$ bash vis.sh
Namespace(decoder_depth=4, device='cuda:0', drop_path=0.0, imagenet_default_mean_and_std=True, img_path='./input/test1.mp4', input_size=224, mask_ratio=0.9, mask_type='tube', model='pretrain_videomae_base_patch16_224', model_path='./ckp/ucf101.pth', num_frames=16, sampling_rate=4, save_path='.demo')

Creating model: pretrain_videomae_base_patch16_224
Patch size = (16, 16)
Traceback (most recent call last):
File "run_videomae_vis.py", line 185, in
main(opts)
File "run_videomae_vis.py", line 102, in main
model.load_state_dict(checkpoint['model'])
KeyError: 'model'

i wonder how can i solve this question ?

你好,请问你解决这个问题了吗,我今天在运行.sh的时候也发生了这样的错误。期待你的回复。
Hello, did you fix this problem, I also got this error when I was running the .sh today. Looking forward to hearing from you.