Struct MomentumSGDArgs

pub struct MomentumSGDArgs<'a, T> { /* private fields */ }

Defines the list passed to the cuda kernel function as arguments to the Momentum SGD optimizer.

Implementations

impl<'a, T> MomentumSGDArgs<'a, T>

Create an instance of an object representing the argument list of the Momentum SGD optimizer.

pub fn new( weight: &'a mut CudaMutPtr<'a, CudaMemoryPoolPtr<T>>, grad: &'a CudaMemoryPoolPtr<T>, size: usize, a: T, mu: T, weight_decay: T, vt: &'a mut CudaMemoryPoolPtr<T>, ) -> MomentumSGDArgs<'a, T>

Create a MomentumSGDArgs instance

Arguments

• weight - unit weight
• grad - gradient
• size - number of weights to be updated
• a - learning rate
• mu - mu
• weight_decay - Weight decay
• vt - vt

Trait Implementations

impl<'a, T> KernelArgs for MomentumSGDArgs<'a, T>

where T: AsKernelPtr,

fn as_vec(&mut self) -> Vec<&mut dyn AsKernelPtr>

Returns a Vec<&mut dyn AsMutKernelPtr> of the type implementing AsMutKernelPtr, which is converted to a data type that can be passed to the cuda kernel in subsequent processing.