未验证 提交 1ab5c511 编写于 作者: S ShenLiang 提交者: GitHub

fix bugs in fp16 for dp (#38405)

上级 49216134
......@@ -356,7 +356,8 @@ def sync_params_buffers(model,
src_rank=0,
is_model_parallel=False):
model_vars = []
for _, param in model.state_dict().items():
params_buffers = model.parameters() + model.buffers()
for param in params_buffers:
if not isinstance(param, core.VarBase):
raise TypeError("The data type of '%s' must be Varbase" %
param.name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册