frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Heyang Qin
2023-07-11 17:14:03 -07:00
committed by GitHub
parent e292343d7b
commit e59f69a8ff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
deepspeed/runtime/zero/partition_parameters.py
View File

@ -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)