saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[distributed] Remove unused variable in test_composability/test_pp_composability.py (#143191)

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143191
Approved by: https://github.com/mori360
This commit is contained in:
Tom Ritchford 2024-12-13 12:55:25 +00:00 committed by PyTorch MergeBot
parent de484134e4
commit 91bf2e16de
1 changed files with 0 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
test/distributed/_composable/test_composability/test_pp_composability.py
View file

@ -537,12 +537,6 @@ class ComposabilityTest(MultiProcessTestCase):
else:
pipeline_schedule.step()
# accumulate losses across pipeline microbatches
loss = (
torch.mean(torch.stack(losses))
if is_last_stage
else torch.Tensor([-1.0])
)
for optimizer in optimizers:
optimizer.step()