sparse_adam.pyi 218 B

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