Save and load model

Misc.
1

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

1 Like
2

yes you can save to and load from the param store

1 Like
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!