A library of helpers for PyTorch.

Michaël Gharbi <[email protected]>

From pip: pip install torch-tools

From source: python setup.py install

To quickly get up and running, use our repo initialization script. In your command line, simple run:

ttools.new

This will prompt you to give a name to your new project, and create the necessary files and folders for you.

The documentation webpage can be found here https://torch-tools.readthedocs.io/en/latest/

For a simple demo, look at the MNIST example in examples/train_mnist.py:

python examples/train_mnist.py data out

Dima Smirnov implemented the tensorboard hooks and callbacks.