Struct TernaryConfig 

pub struct TernaryConfig {
    pub accumulation_steps: usize,
    pub ternary_threshold: f32,
    pub scale_learning_rate: f32,
    pub use_stochastic_rounding: bool,
}

Configuration for ternary gradient accumulation.

Example

use vsa_optim_rs::TernaryConfig;

let config = TernaryConfig::default()
    .with_accumulation_steps(8)
    .with_stochastic_rounding(true);

Fields

accumulation_steps: usize

Number of gradient accumulation steps before optimizer update.

ternary_threshold: f32

Threshold for ternary quantization (relative to mean abs).

scale_learning_rate: f32

Learning rate for scale parameters.

use_stochastic_rounding: bool

Whether to use stochastic rounding (unbiased) or deterministic.

Implementations

impl TernaryConfig

pub const fn with_accumulation_steps(self, steps: usize) -> TernaryConfig

Set the number of accumulation steps.

pub const fn with_stochastic_rounding(self, stochastic: bool) -> TernaryConfig

Set whether to use stochastic rounding.

pub const fn with_threshold(self, threshold: f32) -> TernaryConfig

Set the ternary threshold.

Trait Implementations

impl Clone for TernaryConfig

fn clone(&self) -> TernaryConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TernaryConfig

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

Formats the value using the given formatter.

impl Default for TernaryConfig

fn default() -> TernaryConfig

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for TernaryConfig

fn deserialize<__D>( __deserializer: __D, ) -> Result<TernaryConfig, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for TernaryConfig

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.