Skip to content

Commit

Permalink
resolve test_multiple_optimizers_manual_return
Browse files Browse the repository at this point in the history
  • Loading branch information
tchaton authored and williamFalcon committed Oct 20, 2020
1 parent 77fae0e commit b47b390
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tests/trainer/optimization/test_manual_optimization.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,8 +117,6 @@ def training_step(self, batch, batch_idx, optimizer_idx):
opt_b.zero_grad()
assert torch.all(self.layer.weight.grad == 0)

return {'something': 'else'}

def training_epoch_end(self, outputs) -> None:
# outputs should be an array with an entry per optimizer
assert len(outputs) == 2
Expand Down

0 comments on commit b47b390

Please sign in to comment.