train.sh与test.sh文件编写
无初始化训练
从iteration 0开始训练:
mkdir -p LOG_TRAIN
GLOG_log_dir="./LOG_TRAIN/" ../../build/tools/caffe.bin train --solver=solver.prototxt --gpu=0
GLOG_log_dir:日志输出文件夹路径。
接着是caffe训练工具的位置。
solver:训练文件所在位置。
gpu:设置使用哪个gpu。
有初始化训练
使用snapshot。当训练过程中断,你想继续运行数据学习,此时只需要调用.solverstate文件即可。:
GLOG_log_dir="./LOG_TRAIN/" ../