0.0.5
TorchEval (0.0.5)
Installation from pypi is recommended:
pip install torcheval
A library with simple and straightforward tooling for model evaluations and a delightful user experience. At a high level TorchEval:
-
Contains a rich collection of high performance metric calculations out of the box. We utilize vectorization and GPU acceleration where possible via PyTorch.
-
Integrates seamlessly with distributed training and tools using torch.distributed
-
Is designed with extensibility in mind: you have the freedom to easily create your own metrics and leverage our toolkit.
-
Provides tools for profiling memory and compute requirements for PyTorch based models.