【发布时间】:2016-09-29 10:23:57
【问题描述】:
我想在我的数据集上微调 bvlc_reference_caffenet 模型。
我做了以下步骤:
我为我拥有的新数据集创建了 lmdb 文件。
我下载了预训练的
bvlc_reference_caffenet.caffemodel。我改变了最后一层,即
"fc8",全连接层:
取了train_val.prototxt,把最后一层"fc8"改成"fc8_tune"。我将
num_output更改为尝试预测的正确输出类数(即 2)。我根据我的数据更改了
solver.prototxt-
我跑
$TOOLS/caffe train --solver=solver.prototxt --weights=bvlc_reference_caffenet.caffemodel
但我收到以下错误
[libprotobuf ERROR google/protobuf/text_format.cc:245] Error parsing text-format caffe.NetParameter: 363:19: String literals cannot cross line boundaries. [libprotobuf ERROR google/protobuf/text_format.cc:245] Error parsing text-format caffe.NetParameter: 363:18: Message type "caffe.LayerParameter" has no field named "fc8_tune". F0531 17:13:33.284981 3670 upgrade_proto.cpp:79] Check failed: ReadProtoFromTextFile(param_file, param) Failed to parse NetParameter file: /caffe-master/models/bvlc_reference_caffenet/train_val.prototxt
谁能帮我解决这个错误?
【问题讨论】:
-
非常感谢...Shai...:)..它正在工作。
标签: machine-learning neural-network protocol-buffers deep-learning caffe