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

Simple answer:

reader = tf.train.NewCheckpointReader(checkpoint_file)

variable1 = reader.get_tensor('layer_name1/layer_type_name')
variable2 = reader.get_tensor('layer_name2/layer_type_name')

Now, after modification to these variables, you can assign it back.

layer_name1_var.set_weights([variable1, variable2])