Struct OptimizerConfig 

pub struct OptimizerConfig {
    pub algorithm: MetaAlgorithm,
    pub meta_lr: f64,
    pub n_tasks_per_batch: usize,
    pub max_params_dim: usize,
}

Configuration for MetaLearningOptimizer.

Fields

algorithm: MetaAlgorithm

The meta-learning algorithm to use.

meta_lr: f64

Outer (meta) learning rate.

n_tasks_per_batch: usize

Number of tasks sampled per meta-update batch.

max_params_dim: usize

Maximum allowed parameter dimensionality.

Implementations

impl OptimizerConfig

pub fn default_maml(dim: usize) -> Self

Create a default MAML configuration.

pub fn default_reptile(dim: usize) -> Self

Create a default Reptile configuration.

Trait Implementations

impl Clone for OptimizerConfig

fn clone(&self) -> OptimizerConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OptimizerConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.