Browse Source

Merge remote-tracking branch 'upstream/master'

pull/3/head
taoxiangdong 5 years ago
parent
commit
8f55be6a4a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      parser/caffe/caffe_custom_parser_adapter.cc

+ 1
- 0
parser/caffe/caffe_custom_parser_adapter.cc View File

@@ -85,6 +85,7 @@ Status CaffeCustomParserAdapter::ParseWeights(const Message *op_src, ge::NodePtr
bool bias_en = false;
bool update_in_turn = (static_cast<int64_t >(op->GetAllInputsSize()) == (layer->bottom_size() + layer->blobs_size()));
int start_pos = layer->bottom_size();
bool update_in_turn = (static_cast<int64_t>(op->GetAllInputsSize()) == (layer->bottom_size() + layer->blobs_size()));
for (int i = 0; i < layer->blobs_size(); ++i) {
ge::GeTensorPtr weight = ge::parser::MakeShared<ge::GeTensor>();
GE_CHECK_NOTNULL(weight);


Loading…
Cancel
Save