experiment.ipynb

Kernel: Python 3

DCGAN

This is an experiment training DCGAN model.

Install the labml-nn package

!pip install labml-nn

Imports

from labml import experiment
from labml_nn.gan.wasserstein.experiment import Configs

Create an experiment

experiment.create(name="mnist_wgan")

Initialize configurations

conf = Configs()

Set experiment configurations and assign a configurations dictionary to override configurations

experiment.configs(conf,
                   {
                       'discriminator': 'cnn',
                       'generator': 'cnn',
                       'label_smoothing': 0.01,
                       'generator_loss': 'wasserstein',
                       'discriminator_loss': 'wasserstein',
                   })

Start the experiment and run the training loop.

with experiment.start():
    conf.run()