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.7.tar.gz
(11.3 kB
view hashes)
Built Distribution
torchzq-1.0.7-py3-none-any.whl
(14.7 kB
view hashes)