如何腌制包装TensorFlow图的python类？

Question

我使用pythontex制作动态可复制的报告。 我有一个包装tensorflow图的类。

在我的文档中，我创建我的类，并在其中构造张量流图。

导入tensorflow模块并构造图是一个漫长的过程，因此我想使结果腌制而不是每次都重新运行该过程。

但是我失败了。 can't pickle class tensorflow.python.util.xxx 。 我想这是一个已知的问题。 如何自定义类以方便地转储和加载它？ 或者也许我可以腌制python会话或tensorflow会话？

Answer 1

尝试使用tf.train.Saver 。 有人在这里写了一篇有用的文章。 您只能腌制在模块顶层定义的类（ tensorflow.xxx可以，但tensorflow.python.util.xxx不能），因此您将无法腌制图形（有关所用内容的详细信息）可以腌，请看这里 ）。 最可能的解决方案是找到另一种方法将其保存到文件中。