Tensorflow: how to save/restore a model?

Question

After you train a model in Tensorflow: How do you save the trained model? How do you later restore this saved model?

Gufran Mirza · Accepted Answer

They built an exhaustive and useful tutorial -> https://www.tensorflow.org/guide/saved_model For Tensorflow >= 2 From the docs: Save Model # Create some variables. v1 = tf.get_variable("v1", shape=[3], initializer = tf.zeros_initializer) v2 = tf.get_variable("v2", shape=[5], initializer = tf.zeros_initializer) inc_v1 = v1.assign(v1+1) dec_v2 = v2.assign(v2-1) # Add an op to initialize the variables. init_op = tf.global_variables_initializer() # Add ops to save and restore all the variables. saver = tf.train.Saver() # Later, launch the model, initialize the variables, do some work, and save the # variables to disk. with tf.Session() as sess: sess.run(init_op) # Do some work with the model. inc_v1.op.run() dec_v2.op.run() # Save the variables to disk. save_path = saver.save(sess, "/tmp/model.ckpt") print("Model saved in path: %s" % save_path) Restore Model tf.reset_default_graph() # Create some variables. v1 = tf.get_variable("v1", shape=[3]) v2 = tf.get_variable("v2", shape=[5]) # Add ops to save and restore all the variables. saver = tf.train.Saver() # Later, launch the model, use the saver to restore variables from disk, and # do some work with the model. with tf.Session() as sess: # Restore variables from disk. saver.restore(sess, "/tmp/model.ckpt") print("Model restored.") # Check the values of the variables print("v1 : %s" % v1.eval()) print("v2 : %s" % v2.eval())

Ednsquare · Answer

For tensorflow 2.0, it is as simple as # Save the model model.save('path_to_my_model.h5') #To restore model new_model = tensorflow.keras.models.load_model( 'path_to_my_model.h5' )

Tensorflow

Similar Questions

How to print the value of a Tensor object in TensorFlow?

Can't download CuDNN. How to download on Linux?

tf.gradients returns None for gradient with self for integer variable?

On a mission to build Next-Gen Community Platform for Developers

Facebook

Twitter

Linkedin

Instagram