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.0.22
Filename, size File type Python version Upload date Hashes
Filename, size torch_tools-0.0.22-py3-none-any.whl (37.0 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size torch-tools-0.0.22.tar.gz (27.5 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page