Struct MemoryEfficientLBFGS 

pub struct MemoryEfficientLBFGS { /* private fields */ }

Memory-efficient L-BFGS with improved history management

Implementations

impl MemoryEfficientLBFGS

pub fn new( params: Vec<Arc<RwLock<Tensor>>>, lr: Option<f32>, max_iter: Option<usize>, tolerance_grad: Option<f32>, tolerance_change: Option<f32>, history_size: Option<usize>, memory_config: Option<MemoryConfig>, ) -> Self

pub fn memory_stats(&self) -> HashMap<String, usize>

Get memory usage statistics

pub fn clear_memory(&mut self)

Clear memory pools and history to free up memory

Trait Implementations

impl Optimizer for MemoryEfficientLBFGS

fn step(&mut self) -> OptimizerResult<()>

Perform a single optimization step

fn zero_grad(&mut self)

Zero all gradients

fn get_lr(&self) -> Vec<f32>

Get the current learning rate

fn set_lr(&mut self, lr: f32)

Set the learning rate

fn add_param_group( &mut self, params: Vec<Arc<RwLock<Tensor>>>, options: HashMap<String, f32>, )

Add a parameter group

fn state_dict(&self) -> OptimizerResult<OptimizerState>

Get state dict for serialization

fn load_state_dict(&mut self, state: OptimizerState) -> OptimizerResult<()>

Load state dict