提交 a3f3e830 编写于 作者: J jiangjiajun

fix scope program for export inference

上级 aaf19f7e
...@@ -356,6 +356,7 @@ class BaseAPI: ...@@ -356,6 +356,7 @@ class BaseAPI:
def export_inference_model(self, save_dir): def export_inference_model(self, save_dir):
test_input_names = [var.name for var in list(self.test_inputs.values())] test_input_names = [var.name for var in list(self.test_inputs.values())]
test_outputs = list(self.test_outputs.values()) test_outputs = list(self.test_outputs.values())
with fluid.scope_guard(self.scope):
if self.__class__.__name__ == 'MaskRCNN': if self.__class__.__name__ == 'MaskRCNN':
from paddlex.utils.save import save_mask_inference_model from paddlex.utils.save import save_mask_inference_model
save_mask_inference_model( save_mask_inference_model(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册