使用tensorflow的tfdbg调试代码常见的一个问题

下面这段代码使用tfdbg来进行调试，如果这段代码在命令行中运行没有问题，如果在IDE(例如：pycharm)中则运行报错:error: cbreak() returned ERR。

# -*- coding: utf-8 -*-
import os
import sys
from tensorflow.python import debug as tfdbg
import numpy as np
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

if __name__ == "__main__":
    a = np.array([[1, 1, 0, 1], [0, 0, 0, 0]])
    b = a.reshape(2, 2, 2, 1)
    # b=a.reshape(2,2,1,2)#input batch为2，通道为1时 ，输出果然是2*3*3*2
    filter = tf.Variable(b, dtype=tf.float32, name='filter')

    a1 = np.arange(32)
    b1 = a1.reshape(1, 4, 4, 2)
    # b1=a1.reshape(2,4,4,1) #batch为2
    input = tf.Variable(b1, dtype=tf.float32, name='input')  # 还要加个float32
    op = tf.nn.conv2d(input, filter, strides=[1, 1, 1, 1], padding='VALID', name='op')
    sess = tf.Session()
    sess = tfdbg.LocalCLIDebugWrapperSession(sess)  # 调试步骤b
    sess.add_tensor_filter("has_inf_or_nan", tfdbg.has_inf_or_nan)  # 后面是一个函数
    sess.run(tf.global_variables_initializer())
    # sess.add_tensor_filter("has_inf_or_nan", tfdbg.has_inf_or_nan)#调试步骤c 命令就是run  ，命令有ni 有输入和输出到下一个哪个结点
    # https://www.cnblogs.com/hellcat/articles/7812119.html 调试教程
    bb = (sess.run(op))
    # cc=bb.reshape(2,3,3)
    # dd=bb.reshape(3,3,2)
    ee = sess.run(tf.squeeze(op, name='ee'))
    print(bb.size)

解决方法

使用

sess = tfdbg.LocalCLIDebugWrapperSession(sess,ui_type="readline") 

替换

sess = tfdbg.LocalCLIDebugWrapperSession(sess) 

运行的时候一定要注意是在debug模式下进行。