Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
deepspeed
Commits
4fef478f
Unverified
Commit
4fef478f
authored
Sep 16, 2020
by
Haibin Lin
Committed by
GitHub
Sep 16, 2020
Browse files
Fix a typo in comments (#415)
* Update stage2.py Co-authored-by:
Jeff Rasley
<
jerasley@microsoft.com
>
parent
f5cce75e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
deepspeed/runtime/zero/stage2.py
deepspeed/runtime/zero/stage2.py
+2
-2
No files found.
deepspeed/runtime/zero/stage2.py
View file @
4fef478f
...
...
@@ -517,7 +517,7 @@ class FP16_DeepSpeedZeroOptimizer(object):
see_memory_usage
(
f
"End ipg_epilogue"
)
# resets all partition to no reduced
# sets rem
i
aning grads to the total number of grads in each partition
# sets rema
i
ning 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
}
"
)
###############
I
dependent Partition Gradient ########################
###############
In
dependent 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"
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment