caffe.exe (caffe.bin)用法回顾

本文详细介绍了Caffe命令行工具的各种使用方式,包括训练、测试、设备查询及时间评估等命令,同时还提供了模型定义、权重加载、迭代次数等参数的设置说明。通过实例展示了如何进行MNIST数据集的训练和测试。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

caffe.bin :command line brew

usage : caffe  <command><args>

commands:

  train:         训练或者微调一个网络

  test:        对一个模型打分测试 

  device_query:     显示GPU诊断信息

  time:          评估模型执行时间

 

Flags from tools/caffe.cpp:

  -gpu (可选参数,给定时运行在GPU模式,‘-gpu all’ 则表示运行在所有可用的GPU设备上,此时真正训练批量大小是N×B,N为指定的GPU设备的数目)

  -iterations (循环迭代次数,默认为50)

  -model  (指定模型定义文本名,*.prototext)

  -sighup_effect (当收到SIGHUP信号时要采取的动作,可选项:snapshot.stop,或none,默认是snapshot,即拍快照)

  -sigint_effect (当收到SIGINT信号时要采取的动作,可选项同上,默认为stop)

  -snapshot  (恢复训练时所需要指定上次终止的快照,*.solvestate)

  -solver  (指定求解器文本文件吗,*.prototxt)

  -weights (指定用于微调的与训练权值,*.caffemodel,不可与snapshot同时出现)

 

example mnist:

train: caffe train --solve = example/mnist/lenet_solver.prototxt

test:  caffe test -model example/mnist/lenet_train_test.prototxt -weights example/mnist/lenet_iter_10000.caffemodel -iterations 100.

转载于:https://www.cnblogs.com/LaplaceAkuir/p/6250358.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值