未验证 提交 77bca4de 编写于 作者: W wangguanqun 提交者: GitHub

fix bug in save_inference_model (#37362)

上级 2e758325
...@@ -990,8 +990,8 @@ class TheOnePSRuntime(RuntimeBase): ...@@ -990,8 +990,8 @@ class TheOnePSRuntime(RuntimeBase):
import paddle import paddle
for var in remaining_vars: for var in remaining_vars:
if var.name not in recv_dense_varnames: # if var.name not in recv_dense_varnames:
continue # continue
tensor = var.get_value() tensor = var.get_value()
paddle.save( paddle.save(
tensor, os.path.join(dirname, var.name), use_binary_format=True) tensor, os.path.join(dirname, var.name), use_binary_format=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册