我正在编写的程序涉及在运行时在模型之间切换。
我目前正在使用 Saver 从磁盘中保存/加载模型,如下所示:https://www.tensorflow.org/api_docs/python/state_ops/saving_and_restoring_variables#Saver .
模型相当小,可以存储在内存中,所以我想知道是否有人知道在内存中存储和恢复这些模型而不是将它们保存到磁盘的方法。
我尝试修改 tensorflow 源以将模型保存到内存中,但是 gen_io_ops
似乎是在编译期间生成的。另一种可能的方法是使用内存映射文件。有谁知道更简单的方法吗?
最佳答案
我只会有两个不同的 session ,它们有自己的计算图。或者,您可以在同一个 session 中复制计算图(变量、操作等的两个副本)。然后你会调用 sess.run(comp1 if useCompOne else comp2)
,但是你想设置它。
关于python - 将 tensorflow 模型存储在内存中,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41421117/