在存储器中存储张量流模型

Question

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

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

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

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

Answer 1

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