下面这段代码使用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模式下进行。