当前位置: 首页>>代码示例>>C++>>正文


C++ SolverParameter::clear_train_net方法代码示例

本文整理汇总了C++中SolverParameter::clear_train_net方法的典型用法代码示例。如果您正苦于以下问题:C++ SolverParameter::clear_train_net方法的具体用法?C++ SolverParameter::clear_train_net怎么用?C++ SolverParameter::clear_train_net使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在SolverParameter的用法示例。


在下文中一共展示了SolverParameter::clear_train_net方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。

示例1:

SolverParameter ModelServer<Dtype>::prepare_model() {
  NetParameter net;
  solver->net()->ToProto(&net);

  for (int i = 0; i < net.layer_size(); ++i) {
    LayerParameter& layer = *net.mutable_layer(i);
    layer.clear_blobs();
    if ((layer.type().find("Data") != std::string::npos)
        && (layer.has_remote_data_param())) {
      layer.set_type("RemoteData");

      for (int j = 0; j < layer.top_size(); ++j) {
        *layer.mutable_remote_data_param()->add_shape()
          = blob_shape_by_name(layer.top(j));
      }
    }
  }

  SolverParameter ret = solver->param();
  ret.clear_net();
  ret.clear_net_param();
  ret.clear_test_net();
  ret.clear_test_net_param();
  ret.clear_train_net();
  *ret.mutable_train_net_param() = net;

  return ret;
}
开发者ID:0hm,项目名称:caffe,代码行数:28,代码来源:ModelServer.cpp


注:本文中的SolverParameter::clear_train_net方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。