recitify error log

pull/8209/head
cjh9368 4 years ago
parent 2b1837afa5
commit 7ca49ff9ef

@ -137,7 +137,7 @@ int Flags::Init(int argc, const char **argv) {
} else if (this->quantTypeIn.empty()) {
this->quantType = QuantType_QUANT_NONE;
} else {
std::cerr << "INPUT ILLEGAL: quantType must be AwareTraining|WeightQuant|PostTraining";
std::cerr << "INPUT ILLEGAL: quantType must be WeightQuant|PostTraining";
return RET_INPUT_PARAM_INVALID;
}

@ -41,11 +41,11 @@ bool WeightQuantizer::IsPosNum(const std::string &str) {
STATUS WeightQuantizer::WeightQuantInputCheck(const converter::Flags *config) {
MS_ASSERT(config != nullptr);
if (!WeightQuantizer::IsPosNum(config->quantWeightChannel)) {
MS_LOG(ERROR) << "convWeightQuantChannelThreshold must be valid pos num.";
MS_LOG(ERROR) << "quantWeightChannel must be valid pos num.";
return RET_ERROR;
}
if (!WeightQuantizer::IsPosNum(config->quantWeightSize)) {
MS_LOG(ERROR) << "quantSize must be valid pos num.";
MS_LOG(ERROR) << "quantWeightSize must be valid pos num.";
return RET_ERROR;
}
if (!WeightQuantizer::IsPosNum(config->bitNum) || (config->bitNum != "8" && config->bitNum != "16")) {

Loading…
Cancel
Save