Struct MultiOptimizerTrainer 

pub struct MultiOptimizerTrainer { /* private fields */ }

Multi-optimizer training system

Implementations

impl MultiOptimizerTrainer

pub fn new() -> Self

Create a new multi-optimizer trainer

pub fn add_optimizer( &mut self, name: String, optimizer: Box<dyn Optimizer>, weight: f64, ) -> Result<()>

Add an optimizer with a name

pub fn add_scheduler( &mut self, optimizer_name: String, scheduler: Box<dyn LRScheduler>, ) -> Result<()>

Add a scheduler for an optimizer

pub fn assign_parameters( &mut self, assignments: HashMap<String, String>, ) -> Result<()>

Assign parameters to optimizers

pub fn step( &mut self, parameters: &HashMap<String, Tensor>, gradients: &HashMap<String, Tensor>, ) -> Result<()>

Perform multi-optimizer training step

pub fn step_schedulers(&mut self, epoch: usize) -> Result<()>

Update learning rates using schedulers

pub fn get_stats(&self) -> MultiOptimizerStats

Get training statistics

Trait Implementations

impl Default for MultiOptimizerTrainer

fn default() -> Self

Returns the “default value” for a type.