diff --git a/pytorch_optimizer/base/optimizer.py b/pytorch_optimizer/base/optimizer.py index e4eb610b4..5fe4e8cbb 100644 --- a/pytorch_optimizer/base/optimizer.py +++ b/pytorch_optimizer/base/optimizer.py @@ -154,7 +154,7 @@ def debias(beta: float, step: int) -> float: r"""Adam-style debias correction. Returns `1.0 - beta ** step`. :param beta: float. beta. - :param step. int. number of step. + :param step: int. number of step. """ return 1.0 - math.pow(beta, step) # fmt: skip