Question

使用tf.data.Dataset.from_generator时遇到奇怪的运行时错误。

import tensorflow as tf

def gen():
  while True:
    yield 0, 0

batch = (
  tf.data.Dataset.from_generator(
    gen, output_types=(tf.int32, tf.int32))
  .batch(8)
  .repeat()
  .make_one_shot_iterator()
  .get_next()
)

sess = tf.InteractiveSession()
sess.run(batch)

会打印这样的消息

Exception ignored in: <generator object _yield_value at 0x000001FD9ED6A7D8>
Traceback (most recent call last):
  File "C:\Users\user209974\AppData\Local\Continuum\miniconda3\envs\tf\lib\site-packages\tensorflow\python\data\util\nest.py", line 100, in _yield_value
    yield value
SystemError: error return without exception set

显然，每次调用生成器时，在调用Session.run时都会在运行时引发错误。奇怪的是，尽管抛出此错误，但返回的值似乎是正确的。

那是什么导致此错误，以及如何消除该错误？（由于某些原因，在gen中删除批处理或返回单个值会使错误消失）。

编辑

我意识到在PyCharm下以Debug模式运行时会出现这些错误。如果我在python控制台中以脚本或PyCharm的形式运行代码，但未在Debug模式下运行，则不会打印这些消息。

Answer 1

似乎是Pycharm问题。解决方案是在pycharm设置中设置此环境变量：

PYDEVD_USE_FRAME_EVAL=NO

Check this post that offers more information

使用tf.data.Dataset.from_generator时出现“ SystemError：没有设置异常的错误返回”

1 个答案: