Tensorflow保存模型以进行预测

时间:2018-10-15 13:11:59

标签: tensorflow keras dataset

我试图使用完全使用twnsorflow编写的库张量网络进行转移学习,我的问题是我如何才能像keras一样保存受过训练的网络?

例如,在keras中,我在训练后保存了模型,并加载了模型并使用model.predict之后,如何在tensorlfow中实现呢?

我能够使用Saver对象保存模型,但是它在迭代器和其他方面仍然存在一些问题...实际上,我想将训练阶段和测试阶段完全分开

0 个答案:

没有答案