File size: 306 Bytes
c61ccee
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from .optimizer import Optimizer, ParamsT

class SGD(Optimizer):
    def __init__(
        self,
        params: ParamsT,
        lr: float = ...,
        momentum: float = ...,
        dampening: float = ...,
        weight_decay: float = ...,
        nesterov: bool = ...,
    ) -> None: ...