Browse Source

!14964 fix weight quant conv bug

From: @xutianchun
Reviewed-by: @zhanghaibo5,@hangangqiang
Signed-off-by: @zhanghaibo5
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
d699de4e7f
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      mindspore/lite/tools/converter/quantizer/weight_quantizer.cc

+ 3
- 2
mindspore/lite/tools/converter/quantizer/weight_quantizer.cc View File

@@ -100,8 +100,9 @@ STATUS WeightQuantizer::DoConvQuantize(CNodePtr cnode) {
}

if (param_value->tensor_type() != mindspore::kNumberTypeFloat32) {
MS_LOG(ERROR) << "model weight data type invalid which is " << param_value->tensor_type();
return RET_ERROR;
MS_LOG(WARNING) << cnode->fullname_with_scope() << " weight data type is not fp32 but "
<< param_value->tensor_type();
return RET_OK;
}
auto status = RET_ERROR;
if (type_id_ == kNumberTypeInt8) {


Loading…
Cancel
Save