在内存中存储张量流模型

Question

我正在撰写的程序涉及在运行期间在模型之间切换。

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

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

我尝试修改tensorflow源以将模型保存到内存，但是在编译期间似乎生成了gen_io_ops。另一种可能的方法是使用内存映射文件。有没有人知道更简单的方法？

Answer 1

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