Spaces:
Running
Running
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: ... | |