tensorflow保存和恢复自动编码器

Question

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

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

我看过很多教程，但没有关于此的教程。

是否有关于此

的标准解决方案

非常感谢

Answer 1

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

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