pytorch / tnt

A lightweight library for PyTorch training tools and utilities
https://pytorch.org/tnt/
Other
1.66k stars 271 forks source link
deep-learning machine-learning neural-network python pytorch reinforcement-learning

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: ```buildoutcfg pip install torchtnt ``` Or, alternatively, via conda: ```buildoutcfg 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: ```buildoutcfg pip install git+https://github.com/pytorch/tnt.git@master ``` To update to the latest version from master: ```buildoutcfg pip install --upgrade git+https://github.com/pytorch/tnt.git@master ```