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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for torch_tools-0.0.20-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 57f73ea5accb0f266809917a8bf7d6a46ea72981321c87cdd71b6ad56a93c27d |
|
MD5 | 0a6d392e65b4da56cc7712aaf5ed535d |
|
BLAKE2b-256 | ca7c208f4a8ecd2c7e7112397d0959283e4e9fa0463f014f94fb977f1a2fd0d5 |