search

rosikand / torchplate

🍽 A minimal and simple experiment module for machine learning research workflows in PyTorch.
https://rosikand.github.io/torchplate/
MIT License
3 stars 0 forks source link

Add model serialization #7

Closed rosikand closed 1 year ago

rosikand commented 2 years ago

Note that pickling the actual torchplate.experiment.Experiment subclass won't work.

rosikand commented 2 years ago

https://arxiv.org/pdf/2104.00254.pdf here specifies that one should perform inference and training as separate classes/API's. Essentially, this means that we should serialize the model weights themselves rather than the experiment or model class.