Struct MemoryEfficientOptimizerBuilder 

pub struct MemoryEfficientOptimizerBuilder { /* private fields */ }

Builder for memory-efficient optimizers

Implementations

impl MemoryEfficientOptimizerBuilder

pub fn new() -> Self

pub fn max_memory_gb(self, gb: f32) -> Self

pub fn use_memory_pool(self, use_pool: bool) -> Self

pub fn compress_state(self, compress: bool) -> Self

pub fn lazy_gradients(self, lazy: bool) -> Self

pub fn checkpoint_interval(self, interval: usize) -> Self

pub fn build_adam( self, params: Vec<Arc<RwLock<Tensor>>>, lr: f32, ) -> MemoryEfficientAdam

pub fn build_lbfgs( self, params: Vec<Arc<RwLock<Tensor>>>, lr: f32, ) -> MemoryEfficientLBFGS

Trait Implementations

impl Default for MemoryEfficientOptimizerBuilder

fn default() -> Self

Returns the “default value” for a type.