Skip to content

Commit

Permalink
syntax
Browse files Browse the repository at this point in the history
  • Loading branch information
rohitgr7 committed Jan 14, 2021
1 parent b483b27 commit d2225be
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/models/test_amp.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,8 +196,7 @@ def configure_optimizers__multiple_schedulers(self):
optimizer2 = optim.SGD(self.parameters(), lr=self.learning_rate)
lr_scheduler1 = optim.lr_scheduler.StepLR(optimizer1, 1, gamma=0.1)
lr_scheduler2 = optim.lr_scheduler.StepLR(optimizer2, 1, gamma=0.1)

return [optimizer1, optimizer2], [lr_scheduler1, lr_scheduler2]
return [optimizer1, optimizer2], [lr_scheduler1, lr_scheduler2]

model = CustomModel()
trainer = Trainer(
Expand Down

0 comments on commit d2225be

Please sign in to comment.