保存广泛而深入的学习模型

时间:2017-05-01 05:00:01

标签: tensorflow

我为我的分类任务使用了广泛而深入的学习模型。我已经在训练数据集上对它进行了训练,并希望将其保存,以便我以后可以将其用于预测目的。我尝试过使用:

saver = tf.train.Saver()
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver.save(sess, 'wide_n_deep')

但我得到的只是:

No variables to save

有人可以在这里建议我做错了吗?提前谢谢。

0 个答案:

没有答案