Trait yarnn::optimizer::Optimizer

pub trait Optimizer<N, B: Backend<N>> {
    type Context: OptimizerContext;
    fn update_params(
        &self, 
        backend: &B, 
        ctx: &mut Self::Context, 
        params: &mut B::Tensor, 
        grads: &mut B::Tensor
    );
}

Associated Types

type Context: OptimizerContext

Loading content...

Required methods

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

Loading content...

Implementations on Foreign Types

impl<'a, N, B: Backend<N>, O: Optimizer<N, B>> Optimizer<N, B> for &'a O

type Context = O::Context

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

Loading content...

Implementors

impl<N, B, O> Optimizer<N, B> for WeightDecay<N, B, O> where
    B: Backend<N> + BackendAxpys<N>,
    O: Optimizer<N, B>, 

type Context = O::Context

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

impl<N, B: Backend<N> + BackendAdam<N>> Optimizer<N, B> for Adam<N, B>

type Context = AdamContext<N, B>

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

impl<N, B: Backend<N> + BackendAdam<N>> Optimizer<N, B> for RMSProp<N, B>

type Context = RMSPropContext<N, B>

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

impl<N, B: Backend<N> + BackendScale<N> + BackendAxpy<N> + BackendAdd<N>> Optimizer<N, B> for Sgd<N, B>

type Context = SgdContext<N, B>

fn update_params(
    &self,
    backend: &B,
    ctx: &mut Self::Context,
    params: &mut B::Tensor,
    grads: &mut B::Tensor
)

Loading content...