How to restore variables using CheckpointReader in Tensorflow

Question

I\'m trying to restore some variables from checkpoint file if same variable name is in current model.
And I found that there is some way as in Tensorfow Github

S

Answer 1

tf.train.NewCheckpointReader is a nifty method that creates a CheckpointReader object. CheckpointReader has several very useful methods. The method that would be the most relevant to your question would be get_variable_to_shape_map().

• get_variable_to_shape_map() provides a dictionary with variable names and shapes:

saved_shapes = reader.get_variable_to_shape_map()
print 'fire9/squeeze1x1/kernels:', saved_shapes['fire9/squeeze1x1/kernels']

Please take a look at this quick tutorial below: Loading Variables from Existing Checkpoints