kaolin-wisp
kaolin-wisp copied to clipboard
model.pth How to load ?
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
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)