mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b460c30893
Enable [snake-case-type-alias (PYI042)](https://docs.astral.sh/ruff/rules/snake-case-type-alias/) Link: #110950 Pull Request resolved: https://github.com/pytorch/pytorch/pull/111114 Approved by: https://github.com/albanD
15 lines
341 B
Python
15 lines
341 B
Python
from typing import Tuple
|
|
|
|
from .optimizer import Optimizer, ParamsT
|
|
|
|
class RAdam(Optimizer):
|
|
def __init__(
|
|
self,
|
|
params: ParamsT,
|
|
lr: float = ...,
|
|
betas: Tuple[float, float] = ...,
|
|
eps: float = ...,
|
|
weight_decay: float = ...,
|
|
decoupled_weight_decay: bool = ...,
|
|
) -> None: ...
|