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):
see_memory_usage(f"End ipg_epilogue")
# 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
def reset_partition_gradient_structures(self):
total_partitions = dist.get_world_size(group=self.dp_process_group)
......@@ -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}"
)
###############Idependent Partition Gradient ########################
############### Independent Partition Gradient ########################
def reduce_independent_p_g_buckets_and_remove_grads(self, param, i):
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",
......
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