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
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.7.tar.gz
(19.9 kB
view hashes)
Built Distribution
Close
Hashes for torch_tools-0.0.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6aac8f2f02cdc9bd88987b701f2fe6d331983a025ea9c35490e721d5117d79cb |
|
MD5 | d58a1e576edea7f9930d7c3efc3c24ed |
|
BLAKE2b-256 | a7f7633d6bcb46e4e8540c33bd884ade579e781796df1f16d4bc23304a9f1d51 |