Tensorflow restore 恢复/载入模型.

Posted 2020-10-05

Tensorflow 恢复/载入模型。可以有两种方法：

1、首先利用tensorflow建立和以前一样的图。再利用restore恢复参数。

　　.....定义图......

　　sess=tf.Session()

　　saver=tf.train.Saver()

　　sess.run(tf.global_variables_initializer())

　　saver.restore(sess, weigths_path)

2、或者直接加载以前的图，将saver修改为：

   　sess=tf.Session()

      saver=tf.train.import_meta_graph(graph_path)

　　sess.run(tf.global_variables_initializer())

　　saver.restore(sess, weigths_path)

　　不过第二种方法需要通过tf.get_tensor_by_name和tf.get_operation_by_name来获取以前的tensor和operation