model.pth How to load ?

NVIDIAGameWorks / kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

Other

1.46k stars 131 forks source link

model.pth How to load ? #53

Open jberta1234 opened 2 years ago

jberta1234 commented 2 years ago

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

samuele-bortolato commented 1 year ago

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)