Type Alias State 

pub type State<T = Array> = HashMap<Rc<str>, T>;

Type alias for common optimizer state.

Aliased Type

pub struct State<T = Array> { /* private fields */ }

Trait Implementations

impl OptimizerState for State<AdafactorState>

type UnflattenError = UnflattenError

Error type for unflatten.

fn state_projection( &mut self, ) -> Result<StateProjection<'_>, StateProjectionError>

Declare all required and optional optimizer-state slots.

fn unflatten<I, K>(input: I) -> Result<Self, Self::UnflattenError>

where Self: Sized, I: IntoIterator<Item = (K, Array)>, K: Ord + AsRef<str> + Into<Rc<str>>,

Unflatten an iterator of key-value pairs into the optimizer state.

fn flatten(&mut self) -> Result<Vec<(Rc<str>, &Array)>, StateProjectionError>

Flatten present state entries in stable key order.

fn flatten_mut( &mut self, ) -> Result<Vec<(Rc<str>, &mut Array)>, StateProjectionError>

Flatten mutable present state entries in stable key order.

fn save_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Save the optimizer state to a safetensors file.

fn load_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Load the optimizer state from a safetensors file.

impl OptimizerState for State

type UnflattenError = Infallible

Error type for unflatten.

fn state_projection( &mut self, ) -> Result<StateProjection<'_>, StateProjectionError>

Declare all required and optional optimizer-state slots.

fn unflatten<I, K>(input: I) -> Result<Self, Self::UnflattenError>

where Self: Sized, I: IntoIterator<Item = (K, Array)>, K: Ord + AsRef<str> + Into<Rc<str>>,

Unflatten an iterator of key-value pairs into the optimizer state.

fn flatten(&mut self) -> Result<Vec<(Rc<str>, &Array)>, StateProjectionError>

Flatten present state entries in stable key order.

fn flatten_mut( &mut self, ) -> Result<Vec<(Rc<str>, &mut Array)>, StateProjectionError>

Flatten mutable present state entries in stable key order.

fn save_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Save the optimizer state to a safetensors file.

fn load_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Load the optimizer state from a safetensors file.

impl OptimizerState for State<(Array, Array)>

type UnflattenError = UnflattenError

Error type for unflatten.

fn state_projection( &mut self, ) -> Result<StateProjection<'_>, StateProjectionError>

Declare all required and optional optimizer-state slots.

fn unflatten<I, K>(input: I) -> Result<Self, Self::UnflattenError>

where Self: Sized, I: IntoIterator<Item = (K, Array)>, K: Ord + AsRef<str> + Into<Rc<str>>,

Unflatten an iterator of key-value pairs into the optimizer state.

fn flatten(&mut self) -> Result<Vec<(Rc<str>, &Array)>, StateProjectionError>

Flatten present state entries in stable key order.

fn flatten_mut( &mut self, ) -> Result<Vec<(Rc<str>, &mut Array)>, StateProjectionError>

Flatten mutable present state entries in stable key order.

fn save_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Save the optimizer state to a safetensors file.

fn load_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Load the optimizer state from a safetensors file.