-1
我已经在Tensorflow中训练了一个神经网络模型并将训练结果保存在目录trainlogs
中。如何从张量流中的训练模型中获得测试数据集的预测
在trainlogs
之内,有一个文件checkpoints
。
我认为这是我必须使用的文件来恢复训练的参数,但我不知道如何使用tf.train.Saver()
来完成此操作。
我已经在Tensorflow中训练了一个神经网络模型并将训练结果保存在目录trainlogs
中。如何从张量流中的训练模型中获得测试数据集的预测
在trainlogs
之内,有一个文件checkpoints
。
我认为这是我必须使用的文件来恢复训练的参数,但我不知道如何使用tf.train.Saver()
来完成此操作。
这是一个教程:https://www.tensorflow.org/programmers_guide/saved_model。
从本教程的简短片段:
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess, "/tmp/model.ckpt")