TensorFlow优化在GPU上非确定性地进行

时间:2016-04-13 20:49:30

标签: gpu tensorflow

我使用Google TensorFlow来安装一个小型的2层网络。我注意到当我使用带有AdamOptimizer或RMSPropOptimizer的GPU时,优化会非确定性地进行。即使我使用set_random_seed为图表设定种子,opt也会每次产生不同的权重。

如果我使用其他优化器(例如GradientDescentOptimizer)或将图表配置为在CPU上运行,则不会出现此问题。

有没有人在TensorFlow中遇到过类似的行为,或者对调试问题有任何想法?

我使用TensorFlow 0.7.1和Nvidia Grid K520 GPU。

0 个答案:

没有答案