//saver
import tensorflow as tf
v1=tf.Variable(tf.constant(1.0,shape=[1]),name='v1');
v2=tf.Variable(tf.constant(2.0,shape=[1]),name='v2');
result=v1+v2;
init_op=tf.initialize_all_variables();
saver=tf.train.Saver();
with tf.Session() as sess:
sess.run(init_op);
saver.save(sess,"model/model.ckpt");
//restore
import tensorflow as tf
saver=tf.train.import_meta_graph("model/model.ckpt.meta");
with tf.Session() as sess:
saver.restore(sess,"model/model.ckpt");
print(tf.get_default_graph().get_tensor_by_name("add:0").eval());
TensorFlow断点处理
最新推荐文章于 2022-02-04 18:48:17 发布