版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/guotong1988/article/details/84563787
其中name是文件里的name,
tensor是代码里的tensor。
name不是代码里的name,
tensor不是文件里的tensor。
然后load_vars
是一个{}
loader = tf.train.Saver(load_vars)
save_file="../uncased_L-12_H-768_A-12/bert_model.ckpt"
reader = tf.train.NewCheckpointReader(save_file)
restore_vars = {v.name.split(':')[0]: v for v in tf.trainable_variables() if 'bert' in v.name}
saver = tf.train.Saver(restore_vars)
saver.restore(session, save_file)