Skip to content
Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.
Python
Branch: master
Clone or download
Latest commit ff95fb1 Aug 11, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
docs first commit Aug 11, 2019
examples first commit Aug 11, 2019
pytorch_wrapper first commit Aug 11, 2019
tests first commit Aug 11, 2019
.gitignore first commit Aug 11, 2019
.readthedocs.yaml first commit Aug 11, 2019
LICENSE first commit Aug 11, 2019
README.md first commit Aug 11, 2019
setup.py first commit Aug 11, 2019

README.md

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
You can’t perform that action at this time.