/pytorch-training

Training Helpers for PyTorch that allow writing Object Oriented Training Loops

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

pytorch-training

Training Helpers for PyTorch that allow writing Object Oriented Training Loops

Installation

run pip install -e . in the root of the repository.

Usage

This repo contains an extensible Training Loop abstraction.

For training, you'll need to build your own Updater based on pytorch_training.updater.Updater. You can then use this updater to build a Trainer object that handles the training.