Catalyst

High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather then write another regular train loop.

Break the cycle - use the Catalyst!

Catalyst is compatible with: Python 3.6+. PyTorch 0.4.1+.

API documentation and an overview of the library can be found here.

In the examples folder of the repository, you can find advanced tutorials and Catalyst best practices.

Installation

pip install catalyst

Overview

Catalyst helps you write compact but full-featured DL & RL pipelines in a few lines of code. You get a training loop with metrics, early-stopping, model checkpointing and other features without the boilerplate.

Features

Universal train/inference loop.
Configuration files for model/data hyperparameters.
Reproducibility – even source code will be saved.
Callbacks – reusable train/inference pipeline parts.
Training stages support.
Easy customization.
PyTorch best practices (SWA, AdamW, 1Cycle, FP16 and more).

Structure

DL – runner for training and inference, all of the classic machine learning and computer vision metrics and a variety of callbacks for training, validation and inference of neural networks.
RL – scalable Reinforcement Learning, actor-critic off-policy continuous actions space algorithms and their improvements with distributed training support.
contrib - additional modules contributed by Catalyst users.
data - useful tools and scripts for data processing.

Getting started: 30 seconds with Catalyst

import torch
from catalyst.dl.experiments import SupervisedRunner

# experiment setup
logdir = "./logdir"
n_epochs = 42

# data
loaders = {"train": ..., "valid": ...}

# model, criterion, optimizer
model = Net()
criterion = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters())

# model runner
runner = SupervisedRunner()

# model training
runner.train(
    model=model,
    criterion=criterion,
    optimizer=optimizer,
    loaders=loaders,
    logdir=logdir,
    n_epochs=n_epochs,
    verbose=True
)

Docker

Please see the docker folder for more information and examples.

Contribution guide

We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so without any further discussion. If you plan to contribute new features, utility functions or extensions, please first open an issue and discuss the feature with us.

Please see the contribution guide for more information.

deep-warrior / catalyst