Skip to main content

OptimizationTask

torsh_optim::neural_optimizer

Trait OptimizationTask 

Source 
pub trait OptimizationTask {
    // Required methods
    fn initialize_parameters(
        &self,
        device: &CpuDevice,
    ) -> OptimizerResult<Vec<Tensor>>;
    fn compute_loss_and_gradients(
        &self,
        parameters: &[Tensor],
    ) -> OptimizerResult<(f32, Vec<Tensor>)>;
    fn name(&self) -> &str;
}
Expand description

Trait for optimization tasks used in meta-learning

Required Methods§

Source

fn initialize_parameters( &self, device: &CpuDevice, ) -> OptimizerResult<Vec<Tensor>>

Initialize parameters for this task

Source

fn compute_loss_and_gradients( &self, parameters: &[Tensor], ) -> OptimizerResult<(f32, Vec<Tensor>)>

Compute loss and gradients for given parameters

Source

fn name(&self) -> &str

Get task name/description

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§