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

Confusion with workaround for using a non-locally trained pth file for model_weights #126

Open
pranjshrivastava opened this issue Feb 1, 2023 · 1 comment
Labels
bug Something isn't working

Comments

@pranjshrivastava
Copy link

Hi Alan,
I have used your Detecto framework code frequently on Google Colaboratory and have always successfully trained and used it for detection applications Colab. Recently I tried to train it on Google Colab and downloaded the model_weights.pth file. I used this file to run the detection locally on a Jetson Nano by Nvidia.

I faced the following Runtime error:
Error(s) in loading state_dict for FasterRCNN

I read through the previous issues and the workaround that you provided for buggy load function. However, you seem to have given model_name for a different model.

I'll be grateful if you could help me fix my problem and what model_name i can use for the workaround of load function.

@pranjshrivastava pranjshrivastava added the bug Something isn't working label Feb 1, 2023
@pranjshrivastava pranjshrivastava changed the title Confusion with workaround for using use a non-locally trained pth file for model_weights Confusion with workaround for using a non-locally trained pth file for model_weights Feb 1, 2023
@SutirthaChakraborty
Copy link

Did you find any solution ? @pranjshrivastava

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants