From 6c193662ec408beb5e6b0e37672e778e5c77b231 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 29 Apr 2022 20:54:48 +0800 Subject: [PATCH] fix comment --- deploy/slim/quantization/quant.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/deploy/slim/quantization/quant.py b/deploy/slim/quantization/quant.py index b9fa4110..f7acb185 100755 --- a/deploy/slim/quantization/quant.py +++ b/deploy/slim/quantization/quant.py @@ -161,6 +161,12 @@ def main(config, device, logger, vdl_writer): if config["Global"]["pretrained_model"] is not None: pre_best_model_dict = load_model(config, model) + quanter = QAT(config=quant_config, act_preprocess=PACT) + quanter.quantize(model) + + if config['Global']['distributed']: + model = paddle.DataParallel(model) + # build loss loss_class = build_loss(config['Loss']) @@ -175,12 +181,6 @@ def main(config, device, logger, vdl_writer): if config["Global"]["checkpoints"] is not None: pre_best_model_dict = load_model(config, model, optimizer) - quanter = QAT(config=quant_config, act_preprocess=PACT) - quanter.quantize(model) - - if config['Global']['distributed']: - model = paddle.DataParallel(model) - # build metric eval_class = build_metric(config['Metric']) -- GitLab