c++ - Caffe C++保存网络caffemodel文件

我已经成功构建并培训了audioCaffe演示，但是该演示并不能节省网络。

我在MatLab中找到了使用Python保存网络的文档，但是我找不到有关C++的任何文档。

我认为会有类似net.save("file.caffemodel")的功能，但我尝试过但没有用。

在caffe.cpp的train函数中，我尝试了以下操作:

  if (FLAGS_snapshot.size()) {
    LOG(INFO) << "Resuming from " << FLAGS_snapshot;
    solver.Solve(FLAGS_snapshot);
  } else if (FLAGS_weights.size()) {
    LOG(INFO) << "Finetuning from " << FLAGS_weights;
    solver.net()->CopyTrainedLayersFrom(FLAGS_weights);
    solver.Solve();
  } else {
    solver.Solve();
  }
  solver.save("file.caffemodel")

但是我有一个no method exists error
有任何想法吗？

最佳答案

请尝试这个...

caffe::NetParameter net_param;
net_->ToProto(&net_param);
caffe::WriteProtoToBinaryFile(net_param, caffe_model_path);