From 1c589d6a3a5a53d53a1137a3847816b51d2881dd Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 11 Apr 2022 05:13:56 +0000 Subject: [PATCH] fix slim export model --- deploy/slim/quantization/export_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index bbd291c3..822fd5da 100755 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -133,6 +133,7 @@ def main(): sub_model_save_path, logger) else: save_path = os.path.join(save_path, "inference") + model.eval() export_single_model(quanter, model, infer_shape, save_path, logger) -- GitLab