Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Evaluation Script #7

Closed
ewwnage opened this issue Dec 7, 2021 · 1 comment
Closed

Evaluation Script #7

ewwnage opened this issue Dec 7, 2021 · 1 comment

Comments

@ewwnage
Copy link

ewwnage commented Dec 7, 2021

Hey,

another issue with the evaluation script is that in evaluate.py line 260 the model's state_dict is loaded:
model.load_state_dict(data['netG'])
but there is no netG key to the dictionary.

I've downloaded the pertained fuseformer.pth swell as the i3d_rgb_imagenet.pt

what could be the reason for the missing dict key?

Thanks in advance :)

@ruiliu-ai
Copy link
Owner

the error "KeyError :‘netG'''has occurred“. has bees fixed.
https://github.com/ruiliu-ai/FuseFormer/blob/master/evaluate.py#L260
model.load_state_dict(data['netG']) -> model.load_state_dict(data)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants