pytorch-accelerated
pytorch-accelerated copied to clipboard
Chris-hughes10
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, a...
Hi, I am using the sample code for [timm model training](https://github.com/Chris-hughes10/pytorch-accelerated/blob/main/examples/vision/using_timm_components/all_timm_components.py). There is a mismatch when am accelerating the code with GPU and otherwise. What can be the reason for...
Add type hints to the library. It makes it much easier to work with VSCode since you get the autocompletion. Notes : - type hints are 3.8 style (aka very...
Metadata
Owner
Metadata
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, a...