pub enum Optimizer {
SGD {
momentum: f64,
nesterov: bool,
},
Adam {
beta1: f64,
beta2: f64,
epsilon: f64,
},
AdamW {
beta1: f64,
beta2: f64,
epsilon: f64,
weight_decay: f64,
},
RMSprop {
alpha: f64,
epsilon: f64,
},
AdaGrad {
epsilon: f64,
},
}Expand description
Optimization algorithms
Variants§
SGD
Stochastic Gradient Descent
Adam
Adam optimizer
AdamW
AdamW (Adam with weight decay)
RMSprop
RMSprop optimizer
AdaGrad
AdaGrad optimizer
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Optimizer
impl RefUnwindSafe for Optimizer
impl Send for Optimizer
impl Sync for Optimizer
impl Unpin for Optimizer
impl UnwindSafe for Optimizer
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more