Chris-hughes10/pytorch-accelerated

A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases, and capable of utilizing different hardware options with no code changes required. Docs: https://pytorch-accelerated.readthedocs.io/en/latest/

Language: Python - Size: 397 KB - Last synced at: 6 days ago - Pushed at: 2 months ago - Stars: 184 - Forks: 19