Module gradient_accumulation

scirs2_neural::training

Module gradient_accumulation 

Source
Expand description

Gradient accumulation utilities

Structsยง

GradientAccumulationConfig
Configuration for gradient accumulation
GradientAccumulator
Gradient accumulator for training
GradientStats
Statistics for gradient tracking