Struct Data2VecState 

pub struct Data2VecState {
    pub teacher_params: Vec<f32>,
    pub step: usize,
}

Mutable state that tracks the teacher EMA parameter vector and training step.

The teacher is initialised to a copy of the online (student) parameters and updated each training step via:

  θ_teacher ← m · θ_teacher + (1−m) · θ_online

Fields

teacher_params: Vec<f32>

EMA teacher parameter vector (flat, same length as the student).

step: usize

Training-step counter (incremented by Self::update_teacher).

Implementations

impl Data2VecState

pub fn new(online_params: &[f32]) -> Self

Initialise state by cloning the current online (student) parameters.

The teacher starts as an exact copy of the student so that the very first EMA update moves from a well-defined position.

pub fn update_teacher( &mut self, online_params: &[f32], momentum: f32, ) -> SslResult<()>

Apply the EMA update θ_teacher ← m·θ_teacher + (1−m)·θ_online and increment the step counter.

Errors

• SslError::InvalidMomentum when momentum is outside [0, 1] or non-finite.
• SslError::DimensionMismatch when online_params has a different length than the stored teacher vector.

pub fn teacher(&self) -> &[f32]

Reference to the current teacher parameter vector.

Trait Implementations

impl Clone for Data2VecState

fn clone(&self) -> Data2VecState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Data2VecState

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

Formats the value using the given formatter.