Trait OptimizerBuilder 

pub trait OptimizerBuilder<U, D>
where
    U: UnitValue<U>,
    D: Device<U>,
{
    type Output: Optimizer<U, D>;

    // Required method
    fn build(&self, size: usize) -> Result<Self::Output, OptimizerBuildError>;
}

OptimizerBuilder Definition

Required Associated Types

type Output: Optimizer<U, D>

Required Methods

fn build(&self, size: usize) -> Result<Self::Output, OptimizerBuildError>

Create and return an optimizer

Arguments

• size - Total number of weights to be optimized

Implementors

impl<U> OptimizerBuilder<U, DeviceCpu<U>> for AdagradBuilder<U, DeviceCpu<U>>

where U: UnitValue<U>, Adagrad<U, DeviceCpu<U>>: Optimizer<U, DeviceCpu<U>>,

type Output = Adagrad<U, DeviceCpu<U>>

impl<U> OptimizerBuilder<U, DeviceCpu<U>> for AdamBuilder<U, DeviceCpu<U>>

where U: UnitValue<U>, Adam<U, DeviceCpu<U>>: Optimizer<U, DeviceCpu<U>>,

type Output = Adam<U, DeviceCpu<U>>

impl<U> OptimizerBuilder<U, DeviceCpu<U>> for MomentumSGDBuilder<U, DeviceCpu<U>>

where U: UnitValue<U>, MomentumSGD<U, DeviceCpu<U>>: Optimizer<U, DeviceCpu<U>>,

type Output = MomentumSGD<U, DeviceCpu<U>>

impl<U> OptimizerBuilder<U, DeviceCpu<U>> for RMSpropBuilder<U, DeviceCpu<U>>

where U: UnitValue<U>, RMSprop<U, DeviceCpu<U>>: Optimizer<U, DeviceCpu<U>>,

type Output = RMSprop<U, DeviceCpu<U>>

impl<U> OptimizerBuilder<U, DeviceGpu<U>> for AdagradBuilder<U, DeviceGpu<U>>

where U: UnitValue<U>, DeviceGpu<U>: Device<U>, Adagrad<U, DeviceGpu<U>>: Optimizer<U, DeviceGpu<U>>,

type Output = Adagrad<U, DeviceGpu<U>>

impl<U> OptimizerBuilder<U, DeviceGpu<U>> for AdamBuilder<U, DeviceGpu<U>>

where U: UnitValue<U>, DeviceGpu<U>: Device<U>, Adam<U, DeviceGpu<U>>: Optimizer<U, DeviceGpu<U>>,

type Output = Adam<U, DeviceGpu<U>>

impl<U> OptimizerBuilder<U, DeviceGpu<U>> for MomentumSGDBuilder<U, DeviceGpu<U>>

where U: UnitValue<U>, DeviceGpu<U>: Device<U>, MomentumSGD<U, DeviceGpu<U>>: Optimizer<U, DeviceGpu<U>>,

type Output = MomentumSGD<U, DeviceGpu<U>>

impl<U> OptimizerBuilder<U, DeviceGpu<U>> for RMSpropBuilder<U, DeviceGpu<U>>

where U: UnitValue<U>, DeviceGpu<U>: Device<U>, RMSprop<U, DeviceGpu<U>>: Optimizer<U, DeviceGpu<U>>,

type Output = RMSprop<U, DeviceGpu<U>>

impl<U, D> OptimizerBuilder<U, D> for SGDBuilder<U, D>

where U: UnitValue<U>, D: Device<U>, SGD<U, D>: Optimizer<U, D>,

type Output = SGD<U, D>