TorchZQ: A simple PyTorch experiment runner.
Project description
TorchZQ: A simple PyTorch experiment runner.
Zouqi (『走起』 in Chinese) means "let's go". When you zouqi your experiment, the experiment will go with you.
Installation
Install from PyPI:
pip install torchzq
Install the latest version:
pip install git+https://github.com/enhuiz/torchzq@master
Run an Example
Training
$ zouqi example/config/mnist.yml train
Testing
$ zouqi example/config/mnist.yml test
TensorBoard
$ tensorboard --logdir logs
Supported Features
- Model checkpoints
- Logging
- Gradient accumulation
- Configuration file
- Configuration file inheritance
- TensorBoard
- (c)GAN training (WGAN-GP)
- FP16
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
torchzq-1.0.5.dev0.tar.gz
(11.6 kB
view hashes)
Built Distribution
Close
Hashes for torchzq-1.0.5.dev0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | be5c1d5bc1fcafa0792ddcaf07578d0bd857435e538d6cfd6979a01d1a4ad6e4 |
|
MD5 | 691f281b8d94af86fbda33c60f4cbad0 |
|
BLAKE2b-256 | eee4ecd586a648c00ebf85136fd1a38c220eb2746eb7e652fc801e55d7e1854e |