如何在Tensorflow培訓期間保存兩個不同的檢查點

Question

假設我們有測試/ val/train分裂。在訓練過程中，我們希望保存一些模型檢查點[save_1]，以便稍後重新開始訓練。

此外，我們希望在培訓期間保存另一個模型，以在驗證集[save_2]上顯示最佳性能。完成培訓後，我們使用save_2報告測試數據的性能。

我的問題是，我們如何在TensorFlow培訓期間擁有兩個不同的tf.savers？無論我看到的例子，只保存[save_1]。 指向任何代碼將不勝感激。

謝謝。

Answer 1

通過使用Estimator來包裝您的模型可以得到相當接近的結果。具體看看在RunConfig中保存多個檢查點的選項（你不必扔掉任何東西）。可以結合ValidationMonitor找到最低的驗證錯誤。