[−][src]Struct tch::nn::Optimizer

pub struct Optimizer<T> { /* fields omitted */ }

Methods

impl<T> Optimizer<T>[src]

pub fn zero_grad(&self)[src]

pub fn clip_grad_value(&self, max: f64)[src]

Clips gradient value at some specified maximum value.

pub fn step(&self)[src]

pub fn backward_step(&self, loss: &Tensor)[src]

pub fn backward_step_clip(&self, loss: &Tensor, max: f64)[src]

pub fn set_lr(&mut self, lr: f64)[src]

Sets the optimizer learning rate.

pub fn set_momentum(&mut self, m: f64)[src]

Sets the optimizer momentum.

Trait Implementations

impl<T: Debug> Debug for Optimizer<T>[src]

Auto Trait Implementations

impl<T> !Send for Optimizer<T>

impl<T> !Sync for Optimizer<T>

Blanket Implementations

impl<T, U> Into for T where
    U: From<T>, [src]

impl<T> From for T[src]

impl<T, U> TryFrom for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T> Borrow for T where
    T: ?Sized, [src]

impl<T> Any for T where
    T: 'static + ?Sized, [src]

impl<T> BorrowMut for T where
    T: ?Sized, [src]

impl<T, U> TryInto for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.