Mastering TensorFlow 1.x

Saving and restoring all graph variables with the saver class

We proceed as follows:

1. To use the saver class, first an object of this class is created:

saver = tf.train.Saver()

2. The simplest way to save all the variables in a graph is to call the save() method with the following two parameters: the session object and the path to the file on the disk where the variables will be saved:

with tf.Session() as tfs:   ...   saver.save(tfs,"saved-models/model.ckpt")

3.  To restore the variables, the restore() method is called:

with tf.Session() as tfs:    saver.restore(tfs,"saved-models/model.ckpt")    ...

4. Let us revisit the example from Chapter 1, TensorFlow 101, the code to save the variables in the graph in the simple example is as follows:

# Assume ...