from .optimizer import _params_t, Optimizer | |
class SGD(Optimizer): | |
def __init__(self, params: _params_t, lr: float, momentum: float=..., dampening: float=..., weight_decay:float=..., nesterov:bool=...) -> None: ... | |
from .optimizer import _params_t, Optimizer | |
class SGD(Optimizer): | |
def __init__(self, params: _params_t, lr: float, momentum: float=..., dampening: float=..., weight_decay:float=..., nesterov:bool=...) -> None: ... | |