Enum burn::optim::record::AdaptorRecord

pub enum AdaptorRecord<O, B>
where
    O: SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>,
    B: AutodiffBackend,
{
    V1(AdaptorRecordV1<O, <B as AutodiffBackend>::InnerBackend>),
}

Optimizer adaptor record.

Records are versioned for backward compatibility, so old records can be loaded.

Variants

V1(AdaptorRecordV1<O, <B as AutodiffBackend>::InnerBackend>)

Version 1.

Implementations

impl<O, B> AdaptorRecord<O, B>

where O: SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>, B: AutodiffBackend,

pub fn into_state<const D: usize>( self, ) -> <O as SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>>::State<D>

Converts the record into the optimizer state.

Returns

The optimizer state.

pub fn from_state<const D: usize>( state: <O as SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>>::State<D>, ) -> AdaptorRecord<O, B>

Converts the optimizer state into the record.

Arguments

• state: The optimizer state.

Returns

The record.

Trait Implementations

impl<O, B> Clone for AdaptorRecord<O, B>

where O: SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>, B: AutodiffBackend,

fn clone(&self) -> AdaptorRecord<O, B>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<O, B> Record<B> for AdaptorRecord<O, B>

where O: SimpleOptimizer<<B as AutodiffBackend>::InnerBackend>, B: AutodiffBackend,

type Item<S: PrecisionSettings> = AdaptorRecordItem<O, B, S>

Type of the item that can be serialized and deserialized.

fn into_item<S>(self) -> <AdaptorRecord<O, B> as Record<B>>::Item<S>

where S: PrecisionSettings,

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

fn from_item<S>( item: <AdaptorRecord<O, B> as Record<B>>::Item<S>, device: &<B as Backend>::Device, ) -> AdaptorRecord<O, B>

where S: PrecisionSettings,

Convert the given item into a record.