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

Error loading model for NVGesture rgb images (nv_rgb.pth) #4

Open
ibhattacharya opened this issue Jun 28, 2022 · 2 comments
Open

Error loading model for NVGesture rgb images (nv_rgb.pth) #4

ibhattacharya opened this issue Jun 28, 2022 · 2 comments
Assignees

Comments

@ibhattacharya
Copy link

I am getting an error when loading the provided pretrained model for NVGesture rgb images nv_rgb.pth). I found that some of the model parameters are missing in the 'state_dict' once the checkpoint is loaded. The saved checkpoint for NVGesture RGB model has slightly different self_attention encoder architecture. After modifying the architecture the model file could be loaded but the results doesn't look right.

Could you please reshare/reupload the correct model file? Thank you very much.

@Deusy94 Deusy94 self-assigned this Jun 29, 2022
@Deusy94
Copy link
Collaborator

Deusy94 commented Jun 29, 2022

Hi @ibhattacharya, thank you for reaching out.
It seems the weight file for the rgb split has been uploaded wrongly. I'm looking into it right now, I'll upload the correct one ASAP

@ibhattacharya
Copy link
Author

Thanks @Deusy94 !

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