p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5067 fix post training quant log 

Merge pull request !5067 from xutianchun/quant_0824
This commit is contained in:
mindspore-ci-bot 2020-08-24 19:12:05 +08:00 committed by Gitee
parent 0f69259abf d861a61990
commit bd955c7508
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/tools/anf_exporter/anf_exporter.cc
View File

@ -69,6 +69,10 @@ int AnfExporter::ConvertQuantParam(const std::unique_ptr<schema::MetaGraphT> &me
// activation // activation
auto input_quant_params = primitive->GetInputQuantParams(); auto input_quant_params = primitive->GetInputQuantParams();
auto node_type = (schema::PrimitiveType)primitive->Type(); auto node_type = (schema::PrimitiveType)primitive->Type();
if (input_quant_params.empty()) {
MS_LOG(ERROR) << "node: " << dst_node->name << " input quant params is empty";
return RET_ERROR;
}
for (size_t i = 0; i < input_quant_params.size(); i++) { for (size_t i = 0; i < input_quant_params.size(); i++) {
if (i >= dst_node->inputIndex.size()) { if (i >= dst_node->inputIndex.size()) {
MS_LOG(ERROR) << "node: " << dst_node->name << " input has " << input_quant_params.size() MS_LOG(ERROR) << "node: " << dst_node->name << " input has " << input_quant_params.size()
@ -93,7 +97,10 @@ int AnfExporter::ConvertQuantParam(const std::unique_ptr<schema::MetaGraphT> &me
auto tensor_output = meta_graph->allTensors[output_index].get(); auto tensor_output = meta_graph->allTensors[output_index].get();
auto output_quant_params = primitive->GetOutputQuantParams(); auto output_quant_params = primitive->GetOutputQuantParams();
if (output_quant_params.empty()) { if (output_quant_params.empty()) {
MS_LOG(WARNING) << "node: " << dst_node->name << " output quant params is empty"; if (node_type != schema::PrimitiveType_QuantDTypeCast) {
MS_LOG(ERROR) << "node: " << dst_node->name << " output quant params is empty";
return RET_ERROR;
}
} else { } else {
for (auto output_quant_param : output_quant_params[0]) { for (auto output_quant_param : output_quant_params[0]) {
if (tensor_output->quantParams.empty()) { if (tensor_output->quantParams.empty()) {

4
mindspore/lite/tools/converter/quantizer/quantize_util.cc
View File

@ -168,11 +168,11 @@ STATUS CalQuantizationParams(schema::QuantParamT *quantParam, double mMin, doubl
int quant_max, int quant_min, int num_bits) { int quant_max, int quant_min, int num_bits) {
MS_ASSERT(quantParam != nullptr); MS_ASSERT(quantParam != nullptr);
if (mMin > 0.0f) { if (mMin > 0.0f) {
MS_LOG(ERROR) << "min " << mMin << " is bigger then 0, set to 0, this may course low precision"; MS_LOG(DEBUG) << "min " << mMin << " is bigger then 0, set to 0, this may course low precision";
mMin = 0.0f; mMin = 0.0f;
} }
if (mMax < 0.0f) { if (mMax < 0.0f) {
MS_LOG(ERROR) << "mMax " << mMax << " is smaller than 0, set to 0, this may course low precision"; MS_LOG(DEBUG) << "mMax " << mMax << " is smaller than 0, set to 0, this may course low precision";
mMax = 0.0f; mMax = 0.0f;
} }
if (mMin > mMax) { if (mMin > mMax) {