之前用过Tensorflow的Tensorboard可视化训练过程,感觉很好。所以在用Caffe时也想对训练的过程进行一下可视化。经过一番了解,简单实现了。现将实现过程记录如下:
1、输出训练过程到log 文件
在采用 shell 脚本进行 caffe 训练时,可以输出训练过程到log 文件,如
$CAFFE_ROOT/build/tools/caffe train -solver="solver_train.prototxt" \
-weights="mobilenet_iter_50000.caffemodel" \
-gpu 1,2,3 \
2</