Options
All
  • Public
  • Public/Protected
  • All
Menu

Base class for all gradient based model optimizers.

Hierarchy

Index

Constructors

Properties

Methods

Constructors

constructor

Properties

model

model: Module

parameters

parameters: Variable<any>[]

Methods

Abstract step

  • step(): void
  • Make one optimization step. This assumes that gradients where computed previously by calling backward on a scalar variable before.

    Returns void

zeroGrads

  • zeroGrads(): void
  • Zeros all gradients of the model parameters. This should be called after each optimization step, when the gradients are no longer needed.

    Returns void

Generated using TypeDoc