fix slim export model

2022-04-11 05:13:56 +00:00 · 2022-04-11 05:13:56 +00:00 · 1c589d6a3a
parent d76bc57ab9
commit 1c589d6a3a
1 changed files with 1 additions and 0 deletions
--- 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)