Closed rosikand closed 1 year 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.
Note that pickling the actual
torchplate.experiment.Experiment
subclass won't work.