PyTorch Unsafe torch.load Usage (Static Scan)
pytorch-unsafe-load-usage is a high severity check in the
ai category, written in python.
Its source is templates/ai/ml/torch-unsafe-load-usage.py in
cert-x-gen-templates.
What it checks
Section titled “What it checks”Detects PyTorch torch.load calls that may deserialize untrusted data (missing weights_only=True)
Header
Section titled “Header”| Field | Value |
|---|---|
| Id | pytorch-unsafe-load-usage |
| Severity | high |
| Language | python |
| Category | ai |
| Author | CERT-X-GEN Security Team |
| Template version | not recorded |
| Confidence | 70 |
| CVSS | not recorded |
| Weakness | CWE-502 |
| Tags | ai, ml, pytorch, torch, deserialization, rce, static-scan |
| Target kind | not recorded |
| Oracle | not recorded |
References
Section titled “References”Declared in the header. cxg parses @references and then discards it, and
nothing at scan time reads it, so this is the only place the links a
template cites are surfaced.
Run it
Section titled “Run it”To see what the copy on your machine says about itself, and to confirm it is installed at all:
cxg template info pytorch-unsafe-load-usageThe id it prints is the one to pass anywhere a template is selected. See
cxg template for the rest of
the subcommand, Scan a target for running
a scan, and
A match is not a finding for how
to read what comes back.
Related
Section titled “Related”- Template catalog is the whole corpus.
- ai is the rest of this category.
- Templates by weakness class groups checks by the defect they look for.
- Contribute a template is the route from a check you needed and could not find to a merged one.