Options
All
  • Public
  • Public/Protected
  • All
Menu

Class SGDOptimizer

Hierarchy

Index

Constructors

constructor

  • new SGDOptimizer(learningRate: number, specifiedVariableList?: Node[]): SGDOptimizer

Properties

Protected c

Protected learningRate

learningRate: number

Protected one

one: Scalar<"float32" | "int32" | "bool"> = Scalar.new(1)

Protected prevBatchSize

prevBatchSize: number

Protected specifiedVariableNodes

specifiedVariableNodes: VariableNode[] | null

Protected variableGradients

variableGradients: TensorArrayMap = new TensorArrayMap()

Protected variableNodes

variableNodes: VariableNode[]

Methods

afterBatch

  • afterBatch(math: NDArrayMath, batchSize: number, runtime: SessionRuntime, activationArrayMap: TensorArrayMap, gradientArrayMap: SummedTensorArrayMap): void

afterExample

  • afterExample(math: NDArrayMath, runtime: SessionRuntime, activationArrayMap: TensorArrayMap, gradientArrayMap: SummedTensorArrayMap): void

beforeBatch

  • beforeBatch(math: NDArrayMath, batchSize: number, runtime: SessionRuntime, activationArrayMap: TensorArrayMap, gradientArrayMap: SummedTensorArrayMap): void

dispose

  • dispose(): void

setLearningRate

  • setLearningRate(learningRate: number): void

Generated using TypeDoc