/tnt

A lightweight library for PyTorch training tools and utilities

Primary LanguagePythonOtherNOASSERTION

TNT

TNT is a library for PyTorch training tools and utilities.

build status pypi version pypi nightly version codecov bsd license 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