adamw.pyi 395 B

12345
  1. from typing import Tuple, Optional
  2. from .optimizer import _params_t, Optimizer
  3. class AdamW(Optimizer):
  4. def __init__(self, params: _params_t, lr: float=..., betas: Tuple[float, float]=..., eps: float=..., weight_decay: float=..., amsgrad: bool = ..., *, foreach: Optional[bool] = ..., maximize: bool = ..., capturable: bool = ..., differentiable: bool = ..., fused: bool = ...) -> None: ...