adam.pyi 257 B

12345
  1. from typing import Tuple
  2. from .optimizer import _params_t, Optimizer
  3. class Adam(Optimizer):
  4. def __init__(self, params: _params_t, lr: float=..., betas: Tuple[float, float]=..., eps: float=..., weight_decay: float=..., amsgrad: bool = ...) -> None: ...