如何在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)

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

希望我明白你的问题！