使用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中但不在调试模式下运行代码,则不会打印这些消息。
这似乎是一个Pycharm问题。解决方案是在pycharm设置中设置此环境变量:
PYDEVD_USE_FRAME_EVAL=NO