onnx_diagnostic.torch_models.hghub.model_inputs¶

onnx_diagnostic.torch_models.hghub.model_inputs.compute_model_size(model: Module) → Tuple[int, int][source]¶: Returns the size of the models (weights only) and the number of the parameters.

onnx_diagnostic.torch_models.hghub.model_inputs.filter_out_unexpected_inputs(model: Module, kwargs: Dict[str, Any], verbose: int = 0)[source]¶: Removes input names in kwargs if no parameter names was found in model.forward.