Browse Source

!5223 Fix thread_num no set bug

Merge pull request !5223 from xutianchun/quant_fix
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c694e48fe9
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/lite/tools/converter/quantizer/post_training_quantizer.h

+ 1
- 2
mindspore/lite/tools/converter/quantizer/post_training_quantizer.h View File

@@ -44,11 +44,10 @@ const char kMethodKL[] = "KL";
constexpr int kDefaultBinNumber = 2048; constexpr int kDefaultBinNumber = 2048;


struct ConfigParam { struct ConfigParam {
// ImageFormat imageFormat;
std::string image_path; std::string image_path;
uint32_t batch_count{100}; uint32_t batch_count{100};
std::string method_x{kMethodKL}; std::string method_x{kMethodKL};
uint32_t thread_num;
uint32_t thread_num{1};
}; };


class PostTrainingQuantizer : public Quantizer { class PostTrainingQuantizer : public Quantizer {


Loading…
Cancel
Save