用caffe自带的画图工具,对训练日志进行可视化遇到了一些问题,记录一下。
说一下我的情况。参考这篇博客的方法:http://blog.csdn.net/u013078356/article/details/51154847
1、把这三个文件:
caffe-master/tools/extra/parse_log.sh
caffe-master/tools/extra/extract_seconds.py
master/tools/extra/plot_training_log.py.example
拷贝到日志路径下。
2、
执行 ./parse_log.sh mylog.log
3、
执行 ./plot_training_log.py.example 0 save.png mylog.log
- Notes:
- 1. Supporting multiple logs.
- 2. Log file name must end with the lower-cased ".log".
- Supported chart types:
- 0: Test accuracy vs. Iters
- 1: Test accuracy vs. Seconds
- 2: Test loss vs. Iters
- 3: Test loss vs. Seconds
- 4: Train learning rate vs. Iters
- 5: Train learning rate vs. Seconds
- 6: Train loss vs. Iters
- 7: Train loss vs. Seconds
说一下我的问题,我用的SSD环境训练的模型,日志输出不用额外加命令,训练过程中的输出日志会保存在jog文件夹下。
按照以上步骤我执行了1、2.都没什么问题。
训练过程中,输出的日志,可以正常用0 1 这两个参数画图 其他的参数用不了,
报错:Traceback (most recent call last):
File "./plot_training_log.py", line 191, in <module>
plot_chart(chart_type, path_to_png, path_to_logs)
File "./plot_training_log.py", line 117, in plot_chart
data = load_data(data_file, x, y)
File "./plot_training_log.py", line 88, in load_data
data[1].append(float(fields[field_idx1].strip()))
IndexError: list index out of range
这个问题暂时还没解决。
训练完成时,这个日志正常输出结束,结果连0 1 这两个参数也没法画图了
报同样的类似错误。
查了一下是因为我的学习率策略是用的multi_step。所以在设置的stepvalue的地方训练日志里会输出一句
MultiStep Status: Iteration 10000, step = 1