A library of helpers to train, evaluate and visualize deep nets with PyTorch.
A library of helpers for PyTorch.
Michaël Gharbi <email@example.com>
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:
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.
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for torch_tools-0.1.5-py3-none-any.whl