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

fix bug in save_inference_model (#37362)

上级 2e758325
......@@ -990,8 +990,8 @@ class TheOnePSRuntime(RuntimeBase):
import paddle
for var in remaining_vars:
if var.name not in recv_dense_varnames:
continue
# if var.name not in recv_dense_varnames:
# continue
tensor = var.get_value()
paddle.save(
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.
先完成此消息的编辑!
想要评论请 注册