pub fn save_model<P: AsRef<Path>>( path: P, parameters: HashMap<String, Tensor>, ) -> Result<()>
Simplified save/load functions