A library of helpers to train, evaluate and visualize deep nets with PyTorch.
Project description
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
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
torch-tools-0.0.1.tar.gz
(12.8 kB
view hashes)
Built Distribution
Close
Hashes for torch_tools-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c0eea8aba878a0f74414a92406611194dbdccf4e524f1359de15f4f183e8591d |
|
MD5 | bdc38821959013773b4315f15a3bd1ff |
|
BLAKE2b-256 | 4f87e4b2186a3f5835b5dbbfe61737f169d93c10bea7771208a6a5202ffdbcfc |