在内存中存储张量流模型

2024-01-26

我正在编写的程序涉及在运行时在模型之间进行切换。

我当前正在使用 Saver 从磁盘保存/加载模型,如下所示:https://www.tensorflow.org/api_docs/python/state_ops/ saving_and_restoring_variables#Saver https://www.tensorflow.org/api_docs/python/state_ops/saving_and_restoring_variables#Saver.

这些模型相当小,可以存储在内存中,所以我想知道是否有人知道如何在内存中存储和恢复这些模型,而不是将它们保存到磁盘中。

我尝试修改张量流源以将模型保存到内存中gen_io_ops似乎是在编译时生成的。另一种可能的方法是使用内存映射文件。有谁知道更简单的方法吗?


我只会有两个不同的会话,它们有自己的计算图。或者,您可以在同一会话中复制计算图(变量、操作等的两个副本)。然后你会打电话sess.run(comp1 if useCompOne else comp2),但是您想要设置它。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在内存中存储张量流模型 的相关文章

随机推荐