Struct PredictorHead 

pub struct PredictorHead {
    pub in_dim: usize,
    pub hidden_dim: usize,
    pub out_dim: usize,
    pub w1: Vec<f32>,
    pub b1: Vec<f32>,
    pub w2: Vec<f32>,
    pub b2: Vec<f32>,
}

2-layer MLP predictor head used by BYOL / SimSiam.

Fields

in_dim: usize

Input dim.

hidden_dim: usize

Hidden dim.

out_dim: usize

Output dim (same as projector output for both BYOL and SimSiam).

w1: Vec<f32>

First layer weights [hidden × in].

b1: Vec<f32>

First layer bias.

w2: Vec<f32>

Second layer weights [out × hidden].

b2: Vec<f32>

Second layer bias.

Implementations

impl PredictorHead

pub fn new( in_dim: usize, hidden_dim: usize, out_dim: usize, rng: &mut LcgRng, ) -> SslResult<Self>

New predictor with Kaiming-init weights.

Errors

SslError::InvalidProjectorDim if any dim is zero.

pub fn forward(&self, x: &[f32]) -> SslResult<Vec<f32>>

Forward pass on a single feature vector.

Errors

SslError::DimensionMismatch if x.len() != self.in_dim.

Trait Implementations

impl Clone for PredictorHead

fn clone(&self) -> PredictorHead

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PredictorHead

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

Formats the value using the given formatter.