Browse Source

!14967 fix weight quant conv bug

From: @xutianchun
Reviewed-by: @HilbertDavid,@ivss
Signed-off-by: @HilbertDavid
pull/14967/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
1d501a85c4
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/lite/tools/converter/quantizer/weight_quantizer.cc

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

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

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


Loading…
Cancel
Save