分布式 Tensorflow:检查失败:大小>=0

2024-03-27

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

我的代码可以在 theano 后端运行,但在张量流后端运行失败:

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

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

谢谢你!

- - 更新 - - -

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

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

虽然这个错误看起来像是溢出,但不可能不是在我的电脑上溢出,而是在超级计算机上溢出。

我怀疑它来自分布式计算的张量流上的错误。


我遇到了同样的错误,但在我缩小批量大小之后,Tensorflow 运行正常。

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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

分布式 Tensorflow:检查失败:大小>=0 的相关文章

随机推荐