PyTorchWrapper is a library that provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch. It also provides several ready to use modules and functions for fast model development.
pip install pytorch-wrapper
git clone https://github.com/jkoutsikakis/pytorch-wrapper.git cd pytorch-wrapper pip install .
Docs & Examples
The docs can be found here.
There are also the following examples in notebook format: