Sign in
android
/
platform
/
external
/
pytorch
/
1ea528ef24
/
.
/
torch
/
optim
/
sgd.pyi
blob: 6f8db42d3e1733c47cea20fe19e76f66cb98aaf0 [
file
] [
log
] [
blame
]
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
:
...