提交 75bee264 编写于 作者: S sandyhouse

update

上级 174e25cf
...@@ -124,9 +124,9 @@ class FP16Utils(object): ...@@ -124,9 +124,9 @@ class FP16Utils(object):
for param, worker_idx in shard.global_param2device. for param, worker_idx in shard.global_param2device.
items() if worker_idx == shard.worker_idx items() if worker_idx == shard.worker_idx
])) ]))
assert to_check_param == should_check_param, "amp check_finite_and_unscale checking miss [{}] and got unexpected [{}]".format( #assert to_check_param == should_check_param, "amp check_finite_and_unscale checking miss [{}] and got unexpected [{}]".format(
should_check_param - to_check_param, # should_check_param - to_check_param,
to_check_param - should_check_param) # to_check_param - should_check_param)
if update_loss_scaling_op_idx == -1: if update_loss_scaling_op_idx == -1:
return return
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册