Installation

The most straigtforward way of installing torch-train is via pip

pip install torch-train

If you wish to stay up to date with the latest development version, you can instead download the source code. In this case, make sure that you have all the required dependencies installed.

Dependencies

torch-train requires the following python packages to be installed:

All dependencies should be automatically downloaded if you install torch-train via pip. However, should you want to install these libraries manually, you can install the dependencies using the requirements.txt file

pip install -r requirements.txt

Or you can install these libraries yourself

pip install -U torch