tensorflow保存和恢复自动编码器

Question

我使用tf.layers.dense构建了一个完全连接的自动编码器。我想保存它并只恢复编码器，以获得嵌入输出。

如何使用tf.train.saver仅恢复编码器？因为我想设置恢复模型的不同批量大小，以便只向其中输入一个数据。

我看过很多教程，但没有关于这方面的教程。

对此有什么标准的解决方案吗

非常感谢

Answer 1

如果你不关心内存空间，最简单的方法是保存整个图(编码器和解码器)，当使用它进行预测时，你可以将编码器的最后一层作为fetch参数传递。Tensorflow将只计算到这一点，与只保存编码器相比，您没有任何计算差异。

否则，您可以创建两个图(一个用于编码器，一个用于解码器)，同时训练它们，并将它们训练在一起。但这有点复杂。