tnt icon indicating copy to clipboard operation
tnt copied to clipboard

A lightweight library for PyTorch training tools and utilities

TNT

TNT is a library for PyTorch training tools and utilities.

travis Documentation Status

Installation

TNT can be installed with pip. To do so, run:

pip install torchtnt

If you run into issues, make sure that Pytorch is installed first.

You can also install the latest version from master. Just run:

pip install git+https://github.com/pytorch/tnt.git@master

To update to the latest version from master:

pip install --upgrade git+https://github.com/pytorch/tnt.git@master