You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

13 lines
283 B

5 months ago
from .optimizer import Optimizer, ParamsT
class ASGD(Optimizer):
def __init__(
self,
params: ParamsT,
lr: float = ...,
lambd: float = ...,
alpha: float = ...,
t0: float = ...,
weight_decay: float = ...,
) -> None: ...