Skip to content
/ tnt Public

A lightweight library for PyTorch training tools and utilities

License

Notifications You must be signed in to change notification settings

pytorch/tnt

Repository files navigation

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