Struct RiemannianAdamConfig 

pub struct RiemannianAdamConfig<M, B> {
    pub lr: f64,
    pub beta1: f64,
    pub beta2: f64,
    pub eps: f64,
    pub weight_decay: f64,
    pub amsgrad: bool,
    pub stabilize: Option<usize>,
    /* private fields */
}

Configuration for the Riemannian Adam optimizer.

This optimizer extends the Adam algorithm to work on Riemannian manifolds, following the approach described in “Riemannian adaptive optimization methods” (Bécigneul & Ganea, 2018).

Example

use manopt_rs::prelude::*;

let config = RiemannianAdamConfig::<Euclidean, burn::backend::NdArray>::new()
    .with_lr(0.001)
    .with_beta1(0.9)
    .with_beta2(0.999)
    .with_eps(1e-8)
    .with_amsgrad(true);

Fields

lr: f64

beta1: f64

beta2: f64

eps: f64

weight_decay: f64

amsgrad: bool

stabilize: Option<usize>

Implementations

impl<M, B> RiemannianAdamConfig<M, B>

where M: Manifold<B>, B: Backend,

pub fn new() -> Self

pub fn with_lr(self, lr: f64) -> Self

pub fn with_beta1(self, beta1: f64) -> Self

pub fn with_beta2(self, beta2: f64) -> Self

pub fn with_eps(self, eps: f64) -> Self

pub fn with_weight_decay(self, weight_decay: f64) -> Self

pub fn with_amsgrad(self, amsgrad: bool) -> Self

pub fn with_stabilize(self, stabilize: Option<usize>) -> Self

impl<M, B> RiemannianAdamConfig<M, B>

where M: Manifold<B>, B: Backend,

pub fn init<Back: AutodiffBackend, Mod: AutodiffModule<Back>>( &self, ) -> OptimizerAdaptor<RiemannianAdam<M, Back::InnerBackend>, Mod, Back>

where M: Manifold<Back::InnerBackend>,

Trait Implementations

impl<M: Clone, B: Clone> Clone for RiemannianAdamConfig<M, B>

fn clone(&self) -> RiemannianAdamConfig<M, B>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<M: Debug, B: Debug> Debug for RiemannianAdamConfig<M, B>

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

Formats the value using the given formatter.

impl<M, B> Default for RiemannianAdamConfig<M, B>

where M: Manifold<B>, B: Backend,

fn default() -> Self

Returns the “default value” for a type.