Save and load model


#1

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


Pyro.get_param_store().load() documentation
#2

yes you can save to and load from the param store


#3

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.


#4

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