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
310 B
13 lines
310 B
5 months ago
|
from .optimizer import Optimizer, ParamsT
|
||
|
|
|
||
|
|
class Adagrad(Optimizer):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
params: ParamsT,
|
||
|
|
lr: float = ...,
|
||
|
|
lr_decay: float = ...,
|
||
|
|
weight_decay: float = ...,
|
||
|
|
initial_accumulator_value: float = ...,
|
||
|
|
eps: float = ...,
|
||
|
|
) -> None: ...
|