MLFrameworkConverter

scirs2_io::ml_framework::converters

Trait MLFrameworkConverter 

Source 
pub trait MLFrameworkConverter {
    // Required methods
    fn save_model(&self, model: &MLModel, path: &Path) -> Result<()>;
    fn load_model(&self, path: &Path) -> Result<MLModel>;
    fn save_tensor(&self, tensor: &MLTensor, path: &Path) -> Result<()>;
    fn load_tensor(&self, path: &Path) -> Result<MLTensor>;
}
Expand description

Trait for ML framework converters

Required Methods§

Source

fn save_model(&self, model: &MLModel, path: &Path) -> Result<()>

Source

fn load_model(&self, path: &Path) -> Result<MLModel>

Source

fn save_tensor(&self, tensor: &MLTensor, path: &Path) -> Result<()>

Source

fn load_tensor(&self, path: &Path) -> Result<MLTensor>

Implementors§