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();