Caffe命令参数解析

caffe的c++主程序(caffe.cpp)放在根目录下的tools文件夹内。经过编译后,这些文件都被编译成了可执行文件,放在了 ./build/tools/ 文件夹内。

caffe程序的命令行执行格式如下:caffe <command> <args>

<command>包含命令如下:

train——训练或finetune模型(finetune就是微调的意思

test——测试模型

device_query——显示GPU信息

time——显示程序执行时间

<args>包含的参数有(注意前面有-):

-solver——必选参数。即模型配置文件,为一个protocol buffer(简称PB,谷歌的一种数据交换格式)类型的文件。

-gpu——可选参数。用来指定哪一块gpu运行。设为'-gpu all'则使用所有的gpu运行。

-snapshot——可选参数。用来从快照(snapshot)中恢复训练。

-weights——可选参数。用预先训练好的权重来fine-tuning模型,需要一个caffemodel,不能和-snapshot同时使用。

-iteration——可选参数,迭代次数,默认为50。

-model——可选参数,定义在protocol buffer文件中的模型。

-sighup_effect——可选参数。用来设定当程序发生挂起事件时,执行的操作,可以设置为snapshot, stop或none, 默认为snapshot。

-sigint_effect——可选参数。用来设定当程序发生键盘中止事件时(ctrl+c), 执行的操作,可以设置为snapshot, stop或none, 默认为 stop。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值