如何使用兩個檢查點文件初始化圖形

Question

我已經訓練了兩個模型。他們每個人都可以產生有用的代表特徵。我想用這些功能來訓練第三個模型。這意味着前兩個模型可以產生第三個模型的輸入。我想用tf.contrib.learn.train（）來訓練第三個模型。問題是如何將兩個檢查點文件導入前兩個模型來初始化權重。我知道函數中有init_fn和init_op參數。但是，我不知道如何使用多個檢查點文件。你能幫我嗎？謝謝。

Answer 1

您可以使用tf.contrib.framework.init_from_checkpoint。