Spaces:
Running
Running
File size: 393 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from typing import Tuple
from .optimizer import Optimizer, ParamsT
class NAdam(Optimizer):
def __init__(
self,
params: ParamsT,
lr: float = ...,
betas: Tuple[float, float] = ...,
eps: float = ...,
weight_decay: float = ...,
momentum_decay: float = ...,
decoupled_weight_decay: bool = ...,
) -> None: ...
|