挤压网问题

2024-06-08 21:35:11 发布

您现在位置:Python中文网/ 问答频道 /正文

我想安装压榨网, 这些是说明

SqueezeNet_v1.0/train_val.prototxt          #model architecture
SqueezeNet_v1.0/solver.prototxt             #additional training details (learning rate schedule, etc.)
SqueezeNet_v1.0/squeezenet_v1.0.caffemodel  #pretrained model parameters

运行第一个命令时,我得到了这个错误,我不知道该怎么做?

^{pr2}$

Tags: modelratetrainingetctrainvaldetailsadditional
2条回答

您需要提供'solver.prototxt'作为caffe train的输入参数,而不是{}

train_val.prototxt是拓扑描述。这只适用于model参数,就像在caffe test命令中一样。在

对于caffe train命令,可以使用solver选项(正如您所做的那样),但是您必须提供solver.prototxt文件的名称(很可能是“解算器.prototxt"). 在

添加了澄清

你应该有一个解算器文件,它的开头和下面的一样。确保求解器文件的路径下面的net:路径是准确的。在

net: "/home/prune/caffe/models/googlenet_v1/train_val.prototxt"
test_iter: 1000
test_interval: 4000
display: 50
average_loss: 50
base_lr: 0.01

相关问题 更多 >