Struct burn_core::optim::GradientsAccumulator

pub struct GradientsAccumulator<M> { /* private fields */ }

Accumulate gradients into a single Gradients object.

Implementations

impl<M> GradientsAccumulator<M>

pub fn new() -> Self

Create a new gradients accumulator.

impl<M> GradientsAccumulator<M>

pub fn accumulate<B: AutodiffBackend>( &mut self, module: &M, grads: GradientsParams )where M: AutodiffModule<B>,

Accumulate the given gradients for each parameter in the given module.

pub fn grads(&mut self) -> GradientsParams

Return the accumulated gradients and reset the accumulator state.

Trait Implementations

impl<M> Default for GradientsAccumulator<M>

fn default() -> Self

Returns the “default value” for a type.