Class SGD

Hierarchy (View Summary)

Optimizer
- SGD

Index

Constructors

Properties

defaults params

Methods

Constructors

constructor

new SGD(
    params: Parameter[],
    lr?: number,
    momentum?: number,
    dampening?: number,
    weight_decay?: number,
    nesterov?: boolean,
    maximize?: boolean,
): SGD
Parameters
- params: Parameter[]
- lr: number = 0.001
- momentum: number = 0.0
- dampening: number = 0.0
- weight_decay: number = 0.0
- nesterov: boolean = false
- maximize: boolean = false
Returns SGD
Overrides Optimizer.constructor
- Defined in work/torch/torch/src/optim/optimizers.ts:15

Properties

defaults

defaults: { [key: string]: any }

params

params: Parameter[]

Methods

step

step(): void
Returns void
Overrides Optimizer.step
- Defined in work/torch/torch/src/optim/optimizers.ts:33

zero_grad

zero_grad(): void
Returns void
Inherited from Optimizer.zero_grad
- Defined in work/torch/torch/src/optim/base.ts:12