tensorflow 调试tfdbg

1、执行pip install pyreadline 安装pyreadline

2、修改对应代码如下

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    sess = tfdbg.LocalCLIDebugWrapperSession(sess,ui_type="readline")#添加这个
    sess.add_tensor_filter("has_inf_or_nan", tfdbg.has_inf_or_nan) #添加这个

3 直接运行脚本  tfdbg后输入run  输出如下界面  其中Tensor name为tensorflow变量名 如果没给变量指定 name 无法确定那个是自己查看的变量

 

4 通过 h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64],name="h_pool2_flat") name给h_pool2_flat设置名字为h_pool2_flat

可以在tfdbg中输入 pt h_pool2_flat:0 输出变量值

pt Tensor name:0可以输出指定变量名