Trait dfdx::nn::SaveToSafetensors

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

Something that can be saved to a .safetensors.

All super::Modules in nn implement SaveToSafetensors.

Provided Methods

fn save_safetensors<P: AsRef<Path>>( &self, path: P ) -> Result<(), SafeTensorError>

Save this object into the .safetensors file determined located at path.

Example:

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

Implementors

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