我在 CNN 模型中使用 Tensorflow GPU,我想在每次执行时重现我的结果。
我尝试设置图形级别种子、numpy 随机种子,甚至操作级别种子。但是,它仍然无法重现。
我也用 CPU 试过 Tensorflow;我得到了一个可重复的结果,但它太慢了。
使用带有 GPU 的 Tensorflow 获得可重现结果的任何想法或调试实验建议?
您是否尝试在每次迭代时保存模型的权重?以 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)
然后,您可以在所需的迭代中加载模型的权重
希望我理解你的问题!
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句