mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix slim export model

Browse Source
This commit is contained in:
LDOUBLEV 2022-04-11 05:13:56 +00:00
parent d76bc57ab9
commit 1c589d6a3a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
deploy/slim/quantization/export_model.py
View File

@ -133,6 +133,7 @@ def main():
sub_model_save_path, logger) sub_model_save_path, logger)
else: else:
save_path = os.path.join(save_path, "inference") save_path = os.path.join(save_path, "inference")
model.eval()
export_single_model(quanter, model, infer_shape, save_path, logger) export_single_model(quanter, model, infer_shape, save_path, logger)