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

Quickstart

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.

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.

Files for torch-tools, version 0.1.5
Filename, size File type Python version Upload date Hashes
Filename, size torch_tools-0.1.5-py3-none-any.whl (43.3 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size torch-tools-0.1.5.tar.gz (33.3 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page