DeepSpeed
pipe engine _aggregate_total_loss: more efficient loss concatenation
#4327
Merged
Merged
pipe engine _aggregate_total_loss: more efficient loss concatenation #4327
tjruwase
merged
6 commits into
deepspeedai:master
from
nelyahu:_aggregate_total_loss
_aggregate_total_loss: more efficient loss concatenation
edd0b8ec
tjruwase
approved these changes
on 2023-10-06
Merge branch 'master' into _aggregate_total_loss
edf32ceb
Merge branch 'microsoft:master' into _aggregate_total_loss
75e4ef04
Fixing the if/else block on which the optimization should take place
932ca43f
Merge branch 'master' into _aggregate_total_loss
7d583380
Merge branch 'microsoft:master' into _aggregate_total_loss
1ef033da
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub