kaolin-wisp icon indicating copy to clipboard operation
kaolin-wisp copied to clipboard

model.pth How to load ?

Open jberta1234 opened this issue 3 years ago • 1 comments

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

jberta1234 avatar Sep 27 '22 07:09 jberta1234

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)

samuele-bortolato avatar Jan 18 '23 19:01 samuele-bortolato