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.dcgan import Configs

Create an experiment

experiment.create(name="mnist_dcgan")

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})

Start the experiment and run the training loop.

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