Trait dfdx::nn::LoadFromSafetensors

pub trait LoadFromSafetensors<E: Dtype + SafeDtype, D: Device<E>>: TensorCollection<E, D> {
    // Provided method
    fn load_safetensors<P: AsRef<Path>>(&mut self, path: P) -> Result<(), Error> { ... }
}

Something that can be loaded from a .safetensors file.

All super::Modules in nn implement LoadFromSafetensors.

Provided Methods

fn load_safetensors<P: AsRef<Path>>(&mut self, path: P) -> Result<(), Error>

Loads data from a .safetensors at the specified path.

Example:

let mut model = dev.build_module::<Linear<15, 5>, f32>();
model.load_safetensors("model.safetensors").unwrap();

Implementors

impl<E: Dtype + SafeDtype, D: Device<E>, T: TensorCollection<E, D>> LoadFromSafetensors<E, D> for T