This is a utils package with essential functions that can help avoid boilerplate when using pytorch
Until proper documentation is ready, please refer to this repo for an example of how to use this package.
git clone https://github.com/rishiswethan/pytorch_utils.git
cd pytorch_utils
- Checkout the latest release, or stick to the version that worked best with your project
git checkout tags/v1.1.1
- Install requirements.txt
pip install -r requirements.txt
- Install pytorch 2.0 from here