如何在CNN模型中使用tensorflow-gpu獲得可重現的結果？

Question

我在CNN模型中使用Tensorflow GPU，我想在每次執行時重現我的結果。

我已經嘗試設置圖級別種子，numpy隨機種子，甚至操作級別種子。 但是，它仍然不可重復。

我也嘗試過Tensorflow和CPU; 我有一個可重復的結果，但它太慢了。

任何想法或調試實驗建議使用Tensorflow和GPU獲得可重現的結果？

Answer 1

您是否嘗試在每次迭代時保存模型的權重？ 使用keras回調示例：

filepath = 'weights/' + model.name + '.{epoch:02d}-{acc:.2f}.hdf5'
checkpoint = ModelCheckpoint(filepath, monitor='acc', verbose=0, save_weights_only=True, save_best_only=True, mode='auto', period=1)

然后，您可以在所需的迭代中加載模型的權重

希望我明白你的問題！