本着记录自己学习的脚步,以及明记自己学习的知识点不忘记,今天我会开始更新自己博客。加油
首先,说明一下我的环境:win7(x86)系统,python3.6, TensorFlow版本为“termcolor-1.1.0-py3.6”。
在开始调试之前需要保证系统安装了readline这个python库,但是windows下无法安装这个python库,所以我又下载了pyreadline这个库,我也没有实验单装readline(虽然windows会失败)是否能直接使用,还是建议将两个库都安装一下吧,希望有兴趣的小伙伴帮忙验证一下只装readline或者pyreadline能否调试。
使用pip install --upgrade readline 以及 pip install --upgrade pyreadline就可以安装了。
接下来调试的时候,需要在python代码中引入python的debug包,使用:
from tensorflow.python import debug as tfdbg
这里使用LocalCLIDebugWrapperSession函数,以调试TensorFlow的cifar10的代码为例。
在cifar10_train.py的train函数中的以下代码里加入LocalCLIDebugWrapperSession函数,如下:
with tf.train.MonitoredTrainingSession(
checkpoint_dir=FLAGS.train_dir,
hooks=tf.train.StopAtS