1. 程式人生 > >tensorflow key-value name-tensor 方式從檔案載入模型

tensorflow key-value name-tensor 方式從檔案載入模型

其中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)