save_module

Function save_module 

Source
pub fn save_module<B: Backend>(
    path: impl AsRef<Path>,
    module: &dyn Module<B>,
) -> Result<()>
Expand description

Save a module’s parameters to a .safetensors file using its named_parameters().

use shrew::safetensors;
use shrew::prelude::*;

let linear = Linear::<CpuBackend>::new(3, 2, true, DType::F32, &CpuDevice).unwrap();
safetensors::save_module("linear.safetensors", &linear).unwrap();