save_model

Function save_model 

Source
pub fn save_model<M: Module, P: AsRef<Path>>(model: &M, path: P) -> Result<()>
Expand description

Save a model’s state dictionary to a file.

The format is automatically determined from the file extension.