caffe2训练与测试工具使用

本文介绍了如何使用Caffe2的train.sh和test.sh文件进行深度学习模型的训练和测试。无初始化训练是从头开始,指定日志输出路径和训练配置;有初始化训练则利用已有的.solverstate文件继续训练;测试网络精度时,需提供网络模型文件。
摘要由CSDN通过智能技术生成

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/" ../
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值