From 68b56ba7f8127153be883ff9e54b6ae21326fe5c Mon Sep 17 00:00:00 2001 From: "Jane (Yuan) Xu" <31798555+janeyx99@users.noreply.github.com> Date: Tue, 25 Apr 2023 12:46:45 -0400 Subject: [PATCH] Update the MockOptimizer set_to_none default to match PyTorch (#17463) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Carlos Mocholí --- src/lightning/pytorch/core/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lightning/pytorch/core/optimizer.py b/src/lightning/pytorch/core/optimizer.py index 2c66045aa8..ee1cd45661 100644 --- a/src/lightning/pytorch/core/optimizer.py +++ b/src/lightning/pytorch/core/optimizer.py @@ -379,7 +379,7 @@ class _MockOptimizer(Optimizer): if closure is not None: closure() - def zero_grad(self, set_to_none: Optional[bool] = False) -> None: + def zero_grad(self, set_to_none: Optional[bool] = True) -> None: pass # Do Nothing def __repr__(self) -> str: