Struct LBFGSState 

pub struct LBFGSState<B: Backend> {
    pub history_s: Vec<Tensor<B, 1>>,
    pub history_y: Vec<Tensor<B, 1>>,
    pub d: Option<Tensor<B, 1>>,
    pub t: Option<f64>,
    pub prev_flat_grad: Option<Tensor<B, 1>>,
    pub prev_loss: Option<f64>,
    pub g_iter: usize,
}

L-BFGS optimizer state

Fields

history_s: Vec<Tensor<B, 1>>

Historical displacement vectors

history_y: Vec<Tensor<B, 1>>

Historical gradient difference vectors

d: Option<Tensor<B, 1>>

Search direction

t: Option<f64>

Step size from the previous iteration

prev_flat_grad: Option<Tensor<B, 1>>

Flattened gradient from the previous iteration

prev_loss: Option<f64>

Loss value from the previous iteration

g_iter: usize

Global iteration count

Implementations

impl<B: Backend> LBFGSState<B>

pub fn to_device(self, device: &B::Device) -> Self

Moves all historical tensors to the target device.

Trait Implementations

impl<B: Clone + Backend> Clone for LBFGSState<B>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<B: Backend> Default for LBFGSState<B>

fn default() -> Self

Returns the “default value” for a type.

impl<B: Backend> Record<B> for LBFGSState<B>

type Item<S: PrecisionSettings> = LBFGSStateItem<B, S>

Type of the item that can be serialized and deserialized.

fn into_item<S: PrecisionSettings>(self) -> Self::Item<S>

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S: PrecisionSettings>( item: Self::Item<S>, device: &B::Device, ) -> Self

Convert the given item into a record.