Struct ParameterAverager

pub struct ParameterAverager<A: Float, D: Dimension> { /* private fields */ }

Distributed parameter averager

Implementations

impl<A: Float + ScalarOperand + Debug, D: Dimension> ParameterAverager<A, D>

pub fn new(strategy: AveragingStrategy, num_nodes: usize) -> Self

Create a new parameter averager

pub fn initialize(&mut self, params: &[Array<A, D>]) -> Result<()>

Initialize averager with parameter shapes

pub fn set_node_weight(&mut self, node_id: usize, weight: A) -> Result<()>

Set weight for a specific node

pub fn average_parameters( &mut self, node_parameters: &[(usize, Vec<Array<A, D>>)], ) -> Result<()>

Average parameters from multiple nodes

pub fn get_averaged_parameters(&self) -> &[Array<A, D>]

Get current averaged parameters

pub fn get_averaged_parameters_cloned(&self) -> Vec<Array<A, D>>

Get cloned averaged parameters

pub fn reset(&mut self)

Reset averager state

pub fn step_count(&self) -> usize

Get step count

pub fn num_nodes(&self) -> usize

Get number of nodes

pub fn strategy(&self) -> AveragingStrategy

Get averaging strategy

pub fn is_initialized(&self) -> bool

Check if initialized

Trait Implementations

impl<A: Debug + Float, D: Debug + Dimension> Debug for ParameterAverager<A, D>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.