使用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模式下進行。