Skip to main content

A library of helpers to train, evaluate and visualize deep nets with PyTorch.

Project description

Readme

A library of helpers for PyTorch.

Michaël Gharbi <mgharbi@adobe.com>

Installation

From pip: pip install torch-tools

From source: python setup.py install

Documentation

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

Demo

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

python examples/train_mnist.py data out

Contributors

Dima Smirnov implemented the tensorboard hooks and callbacks.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

torch-tools-0.0.15.tar.gz (22.7 kB view hashes)

Uploaded Source

Built Distribution

torch_tools-0.0.15-py3-none-any.whl (28.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page