我使用tensorflow工具编写了神经网络.
一切正常,现在我想导出我的神经网络的最终权重,以制作一个单一的预测方法.
我怎样才能做到这一点？
解决方法:
您需要在训练结束时使用tf.train.Saver课程保存模型.
在初始化Saver对象时,您需要传递要保存的所有变量的列表.最好的部分是您可以在不同的计算图中使用这些保存的变量！
使用创建Saver对象,
# Assume you want to save 2 variables `v1` and `v2`
saver = tf.train.Saver([v1, v2])
使用tf.Session对象保存变量,
saver.save(sess, 'filename');
当然,您可以添加其他详细信息,例如global_step.
您可以使用restore()函数在将来恢复变量.恢复的变量将自动初始化为这些值.
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 [email protected] 举报，一经查实，本站将立刻删除。
