How to re-create module from loaded model?

[vishnukmd7]

Hi, I'm extremely new to both TF and Sonnet.

I was trying to figure out how to save and load a trained network. I was following what was written in https://github.com/deepmind/sonnet#tensorflow-saved-model , but cant figure out how to regenerate a my_module object, with the weights from the loaded model.

Thanks!

[siAyush]

Hi, You can save the model by this: tf.saved_model.save(my_model, "the_saved_model") And for loading your model back use this: new_model = tf.saved_model.load("the_saved_model")