search

zhijian-liu / torchpack

A neural network training interface based on PyTorch, with a focus on flexibility
https://pypi.org/project/torchpack/
MIT License
61 stars 15 forks source link

save and load #29

Closed chky1997 closed 2 years ago

chky1997 commented 2 years ago

Hi, I'm trying to use torchpack to simplify my model, but I got confused while using the save and load part. Now I can save the .pt file under the checkpoints dir using 'Saver()', but I don't know how to load the .pt file. Could you please give me some advice? Thank you!

zhijian-liu commented 2 years ago

You can simply load the saved checkpoint with torch.load() or io.load (from torchpack.utils).