We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b7ddc4a commit 39a38f3Copy full SHA for 39a38f3
pytorch_optimizer/optimizer/utils.py
@@ -13,11 +13,14 @@
13
14
15
def debias_beta(beta: float, step: int) -> float:
16
- r"""Applies the Adam-style debias correction into beta.
+ r"""Apply the Adam-style debias correction into beta.
17
18
Simplified version of `\^{beta} = beta * (1.0 - beta ** (step - 1)) / (1.0 - beta ** step)`
19
+
20
+ :param beta: float. beta.
21
+ :param step: int. number of step.
22
"""
- return (beta**step - beta) / (beta**step - 1.0)
23
+ return (beta ** step - beta) / (beta ** step - 1.0) # fmt: skip
24
25
26
def is_valid_parameters(parameters: PARAMETERS) -> bool:
0 commit comments