Struct TrainConfig 

pub struct TrainConfig {
    pub batch_size: usize,
    pub n_negatives: usize,
    pub temperature: f32,
    pub learning_rate: f32,
    pub flush_threshold: usize,
}

Configuration for contrastive learning training

Fields

batch_size: usize

Batch size for training

n_negatives: usize

Number of negative samples per positive

temperature: f32

Temperature parameter for contrastive loss

learning_rate: f32

Learning rate for optimization

flush_threshold: usize

Number of updates before flushing to storage

Trait Implementations

impl Clone for TrainConfig

fn clone(&self) -> TrainConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TrainConfig

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

Formats the value using the given formatter.

impl Default for TrainConfig

fn default() -> Self

Returns the “default value” for a type.