在内存中存储张量流模型

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)，但您想进行设置。