tensorflow - 分布式 Tensorflow : check failed: size>=0

Question

我正在使用 keras 2.0.6。 tensorflow 的版本是 1.3.0。

我的代码可以使用 theano 后端运行，但使用 tensorflow 后端失败:

F tensorflow/core/framework/tensor_shape.cc:241] 检查失败:大小 >= 0 (-14428307456 vs. 0)

我想知道是否有人能想到可能导致这种情况的任何可能原因。

谢谢!

- - 更新 - - -

我使用 tensorflow 在我的 PC 上测试了完全相同的代码。它运行完美。

但是，当我在 super 计算机上运行它时，它会抛出此错误。

虽然这个错误看起来像溢出，但它不可能不是在我的 PC 上溢出，而是在 super 计算机上溢出。

我怀疑它来自用于分布式计算的 tensorflow 上的错误。

Answer 1

它出现了同样的错误，但在我缩小批量大小之后它运行正常。

我认为原因是它的 GPU 内存不足。