Struct GradientProcessedOptimizer 

pub struct GradientProcessedOptimizer<T> { /* private fields */ }

Wrapper for optimizers that automatically applies gradient processing.

Implementations

impl<T> GradientProcessedOptimizer<T>

pub fn new(base_optimizer: T, config: GradientProcessingConfig) -> Self

Create a new gradient-processed optimizer.

pub fn with_default_processing(base_optimizer: T) -> Self

Create with default gradient processing configuration.

pub fn gradient_processor(&self) -> &GradientProcessor

Get reference to the gradient processor.

pub fn gradient_processor_mut(&mut self) -> &mut GradientProcessor

Get mutable reference to the gradient processor.

pub fn base_optimizer(&self) -> &T

Get reference to the base optimizer.

pub fn base_optimizer_mut(&mut self) -> &mut T

Get mutable reference to the base optimizer.

Trait Implementations

impl<T: OptimizerState> OptimizerState for GradientProcessedOptimizer<T>

fn zero_grad(&mut self) -> Result<()>

Zero out gradients

fn step(&mut self, parameters: &mut [Tensor]) -> Result<()>

Perform optimization step

fn get_lr(&self) -> f32

Get current learning rate

fn set_lr(&mut self, lr: f32)

Set learning rate

fn state_dict(&self) -> Result<HashMap<String, Tensor>>

Save optimizer state to dictionary

fn load_state_dict(&mut self, state: HashMap<String, Tensor>) -> Result<()>

Load optimizer state from dictionary