Save and load model

Is there an ability to save pretrained probabilistic model and to load its weight distribution’s parameters?

1 Like

yes you can save to and load from the param store

1 Like

Thank you for the tip. You provided the link to Tensorflow but Pyro is based on Pytorch. I found in Pyro documentation that class ParamStoreDict has methods for saving and loading.

oh jeez idk how that happened… fixed the link but glad you were able to find it!