Trait dfdx::nn::LoadFromSafetensors
source · 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> { ... }
}
Expand description
Something that can be loaded from a .safetensors
file.
All super::Modules in nn implement LoadFromSafetensors.