Struct OuterAvg 

pub struct OuterAvg;

Identity outer optimizer: new_global = consensus. Stateless, no momentum, no checkpoint artifact. This is the default and reproduces the pre-outer-optimizer averaging behavior exactly.

Trait Implementations

impl Clone for OuterAvg

fn clone(&self) -> OuterAvg

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for OuterAvg

impl Debug for OuterAvg

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

Formats the value using the given formatter.

impl Default for OuterAvg

fn default() -> OuterAvg

Returns the “default value” for a type.

impl OuterOptimizer for OuterAvg

fn outer_step( &mut self, _prev_global: &[Tensor], consensus: &[Tensor], ) -> Result<Vec<Tensor>>

Transform consensus (this window’s work-weighted average) into the new global parameters, given prev_global (the global adopted at the end of the previous window). Momentum is updated internally.

fn checkpoint_state(&self) -> Option<Vec<Tensor>>

Snapshot the checkpointable outer state (the slow momentum), one tensor per model parameter in Module::parameters() order, for <stem>.outer.fdl. None for stateless variants (OuterAvg), which write no artifact. Returns clones (shallow); the caller serializes them synchronously, so the live state may be overwritten by the next window without racing the detached write.

fn load_checkpoint_state(&mut self, _state: Vec<Tensor>) -> Result<()>

Restore outer state from <stem>.outer.fdl on resume. state is positional, one tensor per parameter (Module::parameters() order). Stateless variants ignore it.

fn resets_inner(&self) -> bool

Whether this variant requires disposable inner optimizer state: the worker, each sync, fully overwrites its params with the new global and resets its inner optimizer (clears momentum, step count). This is DiLoCo’s contract — the inner optimizer is restarted every outer round, which is what makes the outer momentum the canonical, resume-faithful state. false (default) keeps the inner loop continuous (OuterAvg, SlowMomentum).