Type Alias State

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

Type alias for common optimizer state.

Aliased Type

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

Trait Implementations

impl OptimizerState for State<(Array, Array)>

type UnflattenError = UnflattenError

Error type for unflatten.

fn flatten(&self) -> impl Iterator<Item = (Rc<str>, &Array)>

Flatten the optimizer state.

fn flatten_mut(&mut self) -> impl Iterator<Item = (Rc<str>, &mut Array)>

Flatten the mutable optimizer state.

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 save_safetensors(&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<AdafactorState>

type UnflattenError = UnflattenError

Error type for unflatten.

fn flatten(&self) -> impl Iterator<Item = (Rc<str>, &Array)>

Flatten the optimizer state.

fn flatten_mut(&mut self) -> impl Iterator<Item = (Rc<str>, &mut Array)>

Flatten the mutable optimizer state.

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 save_safetensors(&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 flatten(&self) -> impl Iterator<Item = (Rc<str>, &Array)>

Flatten the optimizer state.

fn flatten_mut(&mut self) -> impl Iterator<Item = (Rc<str>, &mut Array)>

Flatten the mutable optimizer state.

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 save_safetensors(&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.