vrishk / pytorch-accelerated Goto Github PK
View Code? Open in Web Editor NEWThis project forked from 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/
License: Apache License 2.0