Struct MixedPrecisionGradientAccumulator 

pub struct MixedPrecisionGradientAccumulator { /* private fields */ }

Gradient accumulator with mixed precision

Implementations

impl MixedPrecisionGradientAccumulator

pub fn new() -> Self

Create new gradient accumulator

pub fn accumulate( &mut self, name: &str, gradients: &MixedPrecisionArray, ) -> Result<()>

Accumulate gradients with mixed precision

pub fn get_averaged_gradients(&self) -> HashMap<String, Array2<Float>>

Get averaged gradients

pub fn clear(&mut self)

Clear accumulated gradients

Trait Implementations

impl Default for MixedPrecisionGradientAccumulator

fn default() -> Self

Returns the “default value” for a type.