Skip to content

Commit 5a8d078

Browse files
committed
update optimizer type hint
1 parent 0656aba commit 5a8d078

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/plugins/precision/native_amp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def pre_optimizer_step(
108108
return False
109109

110110
def post_backward(
111-
self, model: "pl.LightningModule", closure_loss: torch.Tensor, optimizer: torch.optim.Optimizer
111+
self, model: "pl.LightningModule", closure_loss: torch.Tensor, optimizer: Optimizer
112112
) -> torch.Tensor:
113113
ret_val = super().post_backward(model, closure_loss, optimizer)
114114
# unscale here to have it inside the closure before the grad tracking and clipping

0 commit comments

Comments
 (0)