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可以输出指定变量名