Save and load model


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

Pyro.get_param_store().load() documentation

yes you can save to and load from the param store


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!