SGD.ConfigΒΆ

Component: SGD

class SGD.Config[source]

Bases: Optimizer.Config

All Attributes (including base classes)

lr: float = 0.001
momentum: float = 0.0

Default JSON

{
    "lr": 0.001,
    "momentum": 0.0
}