未验证 提交 0d096b3b 编写于 作者: W wuhuachaocoding 提交者: GitHub

update some log. (#51274)

上级 38d233d9
......@@ -539,14 +539,6 @@ class GroupShardedStage2(nn.Layer):
self._has_grad_storage[index] = True
else:
self._param_grads.append(param.name)
print(
"Can not add param: {}, param's shape: {}, param align: {}, grad_storages fill: {}, ".format(
param.name,
param.shape,
self._trainable_param2align[param.name],
self._grad_storages[param.dtype][dst_rank]._fill,
)
)
for dtype in self._grad_storages.keys():
self._grad_storage_list.extend(
......
......@@ -159,7 +159,7 @@ def broadcast_input_data(hcg, *inputs, **kwargs):
v_gpu._share_buffer_to(v)
_broadcast_data_help(v, v.shape, v.dtype, hcg)
else:
logger.error("it doesn't support data type {}".format(type(v)))
logger.warning("it doesn't support data type {}".format(type(v)))
for k, v in kwargs.items():
if isinstance(v, (core.VarBase, core.eager.Tensor)):
......@@ -171,7 +171,7 @@ def broadcast_input_data(hcg, *inputs, **kwargs):
_broadcast_data_help(v, v.shape, v.dtype, hcg)
kwargs[k] = v
else:
logger.error("it doesn't support data type {}".format(type(v)))
logger.warning("it doesn't support data type {}".format(type(v)))
return inputs, kwargs
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册