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.

Sign up for GitHub

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

Jump to bottom

model.pth How to load ? #53

Open
jberta1234 opened this issue Sep 27, 2022 · 1 comment
Open

model.pth How to load ? #53

jberta1234 opened this issue Sep 27, 2022 · 1 comment
Labels
documentation Improvements or additions to documentation good first issue Good for newcomers

Comments

@jberta1234
Copy link

More of a question than an issue.
I see that the app saves "model.pth" which by the looks of it contains the pipeline definition and the trained weights, etc...

Is there an example on how to reload that ? (for instance start the interactive app with it ?)
Thanks

Julien
@orperel orperel added documentation Improvements or additions to documentation good first issue Good for newcomers labels Oct 31, 2022
@samuele-bortolato
Copy link

Just load the pipeline from the saved model instead of creating a new one, then pass that to the trainer

pipeline = torch.load(model_path)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
documentation Improvements or additions to documentation good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@orperel @jberta1234 @samuele-bortolato