Enum dfdx::tensor_ops::Momentum
source · pub enum Momentum {
Classic(f64),
Nesterov(f64),
}
Expand description
Momentum used for crate::optim::Sgd and others
Variants§
Classic(f64)
Momentum that is applied to the velocity of a parameter directly.
Nesterov(f64)
Momentum that is applied to both velocity and gradients. See crate::optim::Sgd nesterov paper for more.
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for Momentum
impl Send for Momentum
impl Sync for Momentum
impl Unpin for Momentum
impl UnwindSafe for Momentum
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