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
pip install torchzq
Run an Example
Training
$ zouqi example/config/mnist.yml train
Testing
$ zouqi example/config/mnist.yml test
TensorBoard
$ zqboard --logdir .
Supported Features
- Model checkpoints
- Logging
- Gradient accumulation
- Configuration file
- Configuration file inheritance
- TensorBoard
- (c)GAN training (WGAN-GP)
- Automatic mixed precision
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.4.tar.gz
(11.9 kB
view hashes)
Built Distribution
torchzq-1.0.4-py3-none-any.whl
(15.5 kB
view hashes)
