tensorflow保存和恢復自動編碼器

Question

我使用tf.layers.dense構建了一個完全連接的自動編碼器。 我想保存它並僅還原編碼器以獲取嵌入輸出。

如何使用tf.train.saver僅還原編碼器？ 因為我想為恢復的模型設置不同的批處理大小，所以只能在其中輸入一個數據。

我看到了很多教程，但是沒有關於此的教程。

有什么標准的解決方案嗎

非常感謝你

Answer 1

如果您不關心內存空間，最簡單的方法是保存整個圖形（編碼器和解碼器），並將其用於預測時，可以將編碼器的最后一層作為fetch參數傳遞。 Tensorflow只會計算到這一點，與僅保存編碼器相比，您沒有任何計算差異。

否則，您可以創建兩個圖（一個用於編碼器，一個用於解碼器）同時訓練它們並一起訓練。 但是，這有點復雜。