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.

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> LoadableModule for Dense<E, I, O>

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> LoadableModule for ScalarScale<E>

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, 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: Float + MatMulImpl, const I: usize> LoadableModule for RMSDiv<E, I>

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

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