Struct tch::COptimizer

impl COptimizer

pub fn adam( lr: f64, beta1: f64, beta2: f64, wd: f64, eps: f64, amsgrad: bool ) -> Result<COptimizer, TchError>

pub fn adamw( lr: f64, beta1: f64, beta2: f64, wd: f64, eps: f64, amsgrad: bool ) -> Result<COptimizer, TchError>

pub fn rms_prop( lr: f64, alpha: f64, eps: f64, wd: f64, momentum: f64, centered: bool ) -> Result<COptimizer, TchError>

pub fn sgd( lr: f64, momentum: f64, dampening: f64, wd: f64, nesterov: bool ) -> Result<COptimizer, TchError>

pub fn add_parameters(&mut self, t: &Tensor, group: usize) -> Result<(), TchError>

pub fn set_learning_rate(&mut self, lr: f64) -> Result<(), TchError>

pub fn set_learning_rate_group( &mut self, group: usize, lr: f64 ) -> Result<(), TchError>

pub fn set_momentum(&mut self, m: f64) -> Result<(), TchError>

pub fn set_momentum_group(&mut self, group: usize, m: f64) -> Result<(), TchError>

pub fn set_weight_decay(&mut self, weight_decay: f64) -> Result<(), TchError>

pub fn set_weight_decay_group( &mut self, group: usize, weight_decay: f64 ) -> Result<(), TchError>

pub fn zero_grad(&self) -> Result<(), TchError>

pub fn step(&self) -> Result<(), TchError>

Trait Implementations§

impl Debug for COptimizer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

impl Drop for COptimizer

fn drop(&mut self)

Executes the destructor for this type. Read more

impl Send for COptimizer

Auto Trait Implementations§

impl UnwindSafe for COptimizer

Blanket Implementations§

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for Twhere T: ?Sized,

const: unstable · source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for Twhere T: ?Sized,

const: unstable · source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

const: unstable · source§

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into for Twhere U: From<T>,

const: unstable · source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Same<T> for T

type Output = T

Should always be Self

impl<T, U> TryFrom for Twhere U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source§

fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>

Performs the conversion.

impl<T, U> TryInto for Twhere U: TryFrom<T>,

type Error = >::Error

The type returned in the event of a conversion error.

const: unstable · source§

fn try_into(self) -> Result<U, >::Error>

Performs the conversion.

impl<V, T> VZip<V> for Twhere V: MultiLane<T>,