提交 1c589d6a 编写于 作者: L LDOUBLEV

fix slim export model

上级 d76bc57a
...@@ -133,6 +133,7 @@ def main(): ...@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册