Function parse_pytorch_model

Source
pub fn parse_pytorch_model(
    file_path: &Path,
) -> Result<HashMap<String, TensorStats>>
Expand description

Parse a PyTorch model file (.pth, .pt) and extract tensor information