Trait LoadableModule 

pub trait LoadableModule {
    // Required methods
    fn save(
        &self,
        path: String,
        dict: &mut HashMap<String, Vec<f64>>,
    ) -> Result<(), LoadSaveError>;
    fn load(
        &mut self,
        path: String,
        dict: &HashMap<String, Vec<f64>>,
    ) -> Result<(), LoadSaveError>;
}

A module who’s parameters can be loaded or saved.

Required Methods

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

Saves the parameters to the given dictionary.

FIXME: We should be storing parameters as their base type, not f64.

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

Loads the parameters from the given dictionary.

FIXME: We should be storing parameters as their base type, not f64.

Dyn Compatibility

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementations on Foreign Types

impl<M1: LoadableModule> LoadableModule for (M1,)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

impl<M2: LoadableModule, M1: LoadableModule> LoadableModule for (M1, M2)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

impl<M3: LoadableModule, M2: LoadableModule, M1: LoadableModule> LoadableModule for (M1, M2, M3)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

impl<M4: LoadableModule, M3: LoadableModule, M2: LoadableModule, M1: LoadableModule> LoadableModule for (M1, M2, M3, M4)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

impl<M5: LoadableModule, M4: LoadableModule, M3: LoadableModule, M2: LoadableModule, M1: LoadableModule> LoadableModule for (M1, M2, M3, M4, M5)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

impl<M6: LoadableModule, M5: LoadableModule, M4: LoadableModule, M3: LoadableModule, M2: LoadableModule, M1: LoadableModule> LoadableModule for (M1, M2, M3, M4, M5, M6)

fn save( &self, path: String, dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

fn load( &mut self, path: String, dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

Implementors

impl LoadableModule for Softmax

impl<E: Dtype + Float + MatMulImpl, const I: usize, const O: usize, A: Module<[E; O], Output = [E; O]> + Default + LoadableModule> LoadableModule for GLU<E, I, O, A>

impl<E: Dtype + MatMulImpl, const I: usize, const O: usize, C: Conv1dKernel<E, Const<I>, Const<O>>> LoadableModule for Conv1d<E, I, O, C>

impl<E: Dtype + MatMulImpl, const I: usize, const O: usize> LoadableModule for Dense<E, I, O>

impl<E: Dtype, const I: usize, M: Default + Module<[E; I]> + LoadableModule> LoadableModule for Residual<E, I, M>

impl<E: Dtype, const I: usize, M: Default + Module<[E; I]>> LoadableModule for LR<E, I, M>

impl<E: Dtype, const I: usize> LoadableModule for Bias1d<E, I>

impl<E: Dtype, const I: usize> LoadableModule for Diag<E, I>

impl<E: Dtype> LoadableModule for ScalarScale<E>

impl<E: Float + MatMulImpl, const I: usize> LoadableModule for RMSDiv<E, I>

impl<E: Float, const I: usize> LoadableModule for Swish<E, I>

impl<E: Float> LoadableModule for Activation<E>