Generating MNIST handwritten digit images by GANs built on PyTorch and visualize/track it with Weights & Biases.

Demo project can be found here.

Clone this repo and install required packages. Use of venv is recommended for isolating the environment.

$ git clone https://github.com/ikr7/wandb-pytorch-gan-mnist-demo.git
$ cd wandb-pytorch-gan-mnist-demo
$ pip install -r requirements.txt

Create an account on Weights & Biases and login from your command line. API keys can be found on User Settings.

$ wandb login <your API key>

Create a config file from params.yaml.template and edit it whatever you want.

$ cp params.yaml.template params.yaml
(edit params.yaml)

Start your training and presto!

$ python src/train.py --params params.yaml