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: