使用tf.data.Dataset.from_generator时出现“SystemError:错误返回,无异常设置”

问题描述 投票:0回答:1

使用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下运行调试模式时会出现这些错误。如果我在python控制台中运行代码,作为脚本,或者在PyCharm中但不在调试模式下运行代码,则不会打印这些消息。

python tensorflow pycharm tensorflow-datasets
1个回答
2
投票

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

PYDEVD_USE_FRAME_EVAL=NO

Check this post that offers more information

© www.soinside.com 2019 - 2024. All rights reserved.