torch-accelerator

Easy to use package for accelerate your pytorch model inference


License
MIT
Install
pip install torch-accelerator==0.0.1

Documentation

TorchAccelerator

PyTorch model wrapper to accelerate models