rusheel86/preflight-ml
Pre-flight checks for PyTorch pipelines. Catch silent failures before they waste your GPU.
View on GitHubTrust Signals
- Scorecard Score
- not yet scored
- Maintenance Recency
- Stale
- License
- None
Inputs
| name | description | required | default |
|---|---|---|---|
| dataloader | Path to Python file that defines a variable named dataloader | yes | — |
| model | Path to Python file that defines a variable named model | no | "" |
| loss | Path to Python file that defines a variable named loss_fn | no | "" |
| val-dataloader | Path to Python file that defines a variable named dataloader (val set) | no | "" |
| python-version | Python version to use | no | 3.11 |
| preflight-version | preflight-ml version to install | no | latest |
Outputs
no outputs