pytorch-wrapper 1.0.0

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.

PyTorchWrapper

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.

Installation

From PyPI

pip install pytorch-wrapper

From Source

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:

1. Two Spiral Task
2. Image Classification Task
3. Tuning Image Classifier
4. Text Classification Task
5. Sequence Classification Task
6. Custom Callback
7. Custom Loss Wrapper
8. Custom Evaluator