PyPI page
Home page
Author:
AKJ7
Summary:
pytest plugin for testing PyTorch models: inference contracts, parity, invariance, training loops, benchmarks
Latest version:
1.1.1
Required dependencies:
pytest
|
tomli
Optional dependencies:
accelerate
|
hypothesis
|
lightning
|
matplotlib
|
onnx
|
onnxruntime
|
onnxscript
|
timm
|
torch
|
torchvision
|
transformers
Downloads last day:
236
Downloads last week:
506
Downloads last month:
0
7-day smoothing