diff --git a/deploy/slim/quantization/README.md b/deploy/slim/quantization/README.md index 42d1ad1adee8519180a143dbfb36dfb5ad0be2cc..f2e92f54a5b456b25445282a38fe30e01fe4fd49 100755 --- a/deploy/slim/quantization/README.md +++ b/deploy/slim/quantization/README.md @@ -4,7 +4,7 @@ ## 概述 -该示例使用PaddleSlim提供的[量化压缩API](https://paddlepaddle.github.io/PaddleSlim/api/quantization_api/)对检测模型进行压缩。 +该示例使用PaddleSlim提供的[量化压缩API](https://paddlepaddle.github.io/PaddleSlim/api/quantization_api/)对OCR模型进行压缩。 在阅读该示例前,建议您先了解以下内容: - [OCR模型的常规训练方法](https://github.com/PaddlePaddle/PaddleOCR/blob/develop/doc/doc_ch/detection.md) diff --git a/deploy/slim/quantization/quant.py b/deploy/slim/quantization/quant.py index 70e15de7b91d7a1d20665cc2f5c4b298ca6a999e..eaeb9f22c792bb59ca734ec4b0b408455a865605 100755 --- a/deploy/slim/quantization/quant.py +++ b/deploy/slim/quantization/quant.py @@ -179,26 +179,7 @@ def main(): program.train_eval_rec_run(config, exe, train_info_dict, eval_info_dict) -def test_reader(): - logger.info(config) - train_reader = reader_main(config=config, mode="train") - import time - starttime = time.time() - count = 0 - try: - for data in train_reader(): - count += 1 - if count % 1 == 0: - batch_time = time.time() - starttime - starttime = time.time() - logger.info("reader:", count, len(data), batch_time) - except Exception as e: - logger.info(e) - logger.info("finish reader: {}, Success!".format(count)) - - if __name__ == '__main__': startup_program, train_program, place, config, train_alg_type = program.preprocess( ) main() -# test_reader()