Enum OptimizerArchitecture 

pub enum OptimizerArchitecture {
    RNN {
        hidden_size: usize,
    },
    LSTM {
        hidden_size: usize,
        n_layers: usize,
    },
    Transformer {
        n_heads: usize,
        d_model: usize,
    },
    GNN {
        n_message_passing_steps: usize,
    },
}

Learned optimizer architectures

Variants

RNN

Recurrent Neural Network

Fields

hidden_size: usize

LSTM

LSTM-based optimizer

Fields

hidden_size: usize

n_layers: usize

Transformer

Transformer-based optimizer

Fields

n_heads: usize

d_model: usize

GNN

Graph Neural Network for parameter relationships

Fields

n_message_passing_steps: usize

Trait Implementations

impl Clone for OptimizerArchitecture

fn clone(&self) -> OptimizerArchitecture

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizerArchitecture

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

Formats the value using the given formatter.