TNT is a library for PyTorch training tools and utilities.
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