Распределенный Tensorflow: проверка не удалась: размер › = 0

Я использую керас 2.0.6. Версия тензорного потока 1.3.0.

Мой код может работать с бэкэндом theano, но не работает с бэкэндом tensorflow:

F tensorflow/core/framework/tensor_shape.cc:241] Ошибка проверки: размер >= 0 (-14428307456 против 0)

Мне было интересно, может ли кто-нибудь подумать о какой-либо возможной причине, которая может вызвать это.

Благодарю вас!

----ОБНОВИТЬ-----

Я протестировал точно такой же код на своем ПК с помощью tensorflow. Он работает отлично.

Однако он выдает эту ошибку, когда я запускаю его на суперкомпьютере.

Хотя эта ошибка выглядит как переполнение, не может быть, чтобы она не переполнялась на моем ПК, а переполнялась на суперкомпьютере.

Я подозреваю, что это происходит из-за ошибки в тензорном потоке для распределенных вычислений.


tensorflow tensorflow-gpu
person volcanofly    schedule 31.07.2017    source источник

Ответы (2)


arrow_upward
3
arrow_downward

вышла та же ошибка, но после этого я уменьшил размер партии.

Я думаю, причина в том, что не хватает памяти графического процессора.

person bai wenjie    schedule 05.09.2017

arrow_upward
0
arrow_downward

Я столкнулся с ошибкой, в моей проблеме ошибка исходит от TF с другим видением.

ошибка решена. модель была обучена в tf 1.15, но заморозила модель в tf 1.13. Когда заморозил в tf 1.15 все ок.

Я думаю, вы можете проверить версию модели.

person colten    schedule 05.01.2021