未验证 提交 e59f69a8 编写于 作者: H Heyang Qin 提交者: GitHub

remove the call to param.ds_tensor from print (#3928)

上级 e292343d
......@@ -1290,7 +1290,7 @@ class Init(InsertPostInitMethodToModuleSubClasses):
def _partition(self, param_list, force=False, has_been_updated=False):
for param in param_list:
print_rank_0(f"Before Partitioning Param {param.ds_id} pri: {param.ds_tensor}", force=False)
print_rank_0(f"Before Partitioning Param {param.ds_id}", force=False)
if self.zero_param_process_group is not None:
self._partition_param_sec(param, has_been_updated=has_been_updated)
self._partition_param(param, has_been_updated=has_been_updated)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册