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

Installation

TNT can be installed with pip:

pip install torchtnt

Or, alternatively, via conda:

conda install -c conda-forge 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