Tensorflow使用经过训练的模型来进行预测初始化错误

时间:2018-01-31 13:43:50

标签: tensorflow initializer checkpoint

我已经在一些数据上训练了一个模型(只是一个简单的分类任务)。之后,我希望使用相同的模型通过单独的函数make_prediction()运行一些预测。

所以目前我的主文件只是:

agent.train(data)

agent.make_predictions(new_data)

tf.global_variables_initializer() with tf.Session() as sess: sess.run(init)

我没有在第二个函数中初始化变量,因此会话与之前的会话不同,但令我惊讶的是,我无法重新打开之前的会话。我是否需要在训练后检查模型,然后每次重新加载?

非常感谢

0 个答案:

没有答案