如何在CNN模型中使用tensorflow-gpu获得可再现的结果？

Question

我正在CNN模型中使用Tensorflow GPU，我想在每次执行时重现我的结果。

我尝试设置图形级种子，numpy随机种子，甚至是操作级种子。但是，它仍然无法再现。

我也尝试过使用CPU进行Tensorflow；我得到了可重现的结果，但是太慢了。

使用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)

然后您可以在所需的迭代中加载模型的权重

希望我很好理解你的问题！