Unverified Commit 4fef478f authored by Haibin Lin's avatar Haibin Lin Committed by GitHub
Browse files

Fix a typo in comments (#415)



* Update stage2.py
Co-authored-by: default avatarJeff Rasley <jerasley@microsoft.com>
parent f5cce75e
...@@ -517,7 +517,7 @@ class FP16_DeepSpeedZeroOptimizer(object): ...@@ -517,7 +517,7 @@ class FP16_DeepSpeedZeroOptimizer(object):
see_memory_usage(f"End ipg_epilogue") see_memory_usage(f"End ipg_epilogue")
# resets all partition to no reduced # resets all partition to no reduced
# sets remianing grads to the total number of grads in each partition # sets remaining grads to the total number of grads in each partition
# set is grad computed to false for all grads in partition # set is grad computed to false for all grads in partition
def reset_partition_gradient_structures(self): def reset_partition_gradient_structures(self):
total_partitions = dist.get_world_size(group=self.dp_process_group) total_partitions = dist.get_world_size(group=self.dp_process_group)
...@@ -617,7 +617,7 @@ class FP16_DeepSpeedZeroOptimizer(object): ...@@ -617,7 +617,7 @@ class FP16_DeepSpeedZeroOptimizer(object):
f"{tag}: elems in_bucket {self.elements_in_ipg_bucket} param {param_elems} max_percent {percent_of_bucket_size}" f"{tag}: elems in_bucket {self.elements_in_ipg_bucket} param {param_elems} max_percent {percent_of_bucket_size}"
) )
###############Idependent Partition Gradient ######################## ############### Independent Partition Gradient ########################
def reduce_independent_p_g_buckets_and_remove_grads(self, param, i): def reduce_independent_p_g_buckets_and_remove_grads(self, param, i):
if self.elements_in_ipg_bucket + param.numel() > self.reduce_bucket_size: if self.elements_in_ipg_bucket + param.numel() > self.reduce_bucket_size:
self.report_ipg_memory_usage("In ipg_remove_grads before reduce_ipg_grads", self.report_ipg_memory_usage("In ipg_remove_grads before reduce_ipg_grads",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment