pub fn estimate_model_memory<B: Backend>(module: &dyn Module<B>) -> usize
Estimate the memory usage of a module’s parameters in bytes.
Accounts for the actual dtype of each parameter tensor.