Table of Contents

Class GpuOptimizerState

Namespace
AiDotNet.Interfaces
Assembly
AiDotNet.dll

Holds the optimizer state buffers for GPU-resident training.

public class GpuOptimizerState
Inheritance
GpuOptimizerState
Inherited Members

Remarks

Different optimizers require different state: - SGD with momentum: velocity buffer - Adam/AdamW: first moment (m) and second moment (v) buffers - RMSprop: squared average buffer - Adagrad: accumulated gradient buffer

Properties

AccumulatedGrad

Accumulated gradient buffer (for Adagrad).

public IGpuBuffer? AccumulatedGrad { get; set; }

Property Value

IGpuBuffer

M

First moment buffer (for Adam-family optimizers).

public IGpuBuffer? M { get; set; }

Property Value

IGpuBuffer

SquaredAvg

Squared average buffer (for RMSprop).

public IGpuBuffer? SquaredAvg { get; set; }

Property Value

IGpuBuffer

V

Second moment buffer (for Adam-family optimizers).

public IGpuBuffer? V { get; set; }

Property Value

IGpuBuffer

Velocity

Velocity buffer (for SGD momentum, NAG, LARS).

public IGpuBuffer? Velocity { get; set; }

Property Value

IGpuBuffer