提交 5acf1366 编写于 作者: Q qingqing01 提交者: Yu Yang

Bug fix in CudnnConvLayer, which will lead to destruction error. (#317)

上级 ee028bb5
......@@ -35,8 +35,8 @@ bool CudnnConvLayer::init(const LayerMap &layerMap,
ProjectionConfig* conf = new ProjectionConfig();
conf->set_type("conv");
conf->set_num_filters(numFilters_);
conf->set_allocated_conv_conf(
config_.mutable_inputs(i)->mutable_conv_conf());
ConvConfig* convConf = conf->mutable_conv_conf();
*convConf = *(config_.mutable_inputs(i)->mutable_conv_conf());
conf->set_input_size(getPrev(i)->getSize());
conf->set_output_size(getSize());
projConf_.emplace_back(conf);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册