Experiment - Annotated Deep Learning Paper Implementations

Denoising Diffusion Probabilistic Models (DDPM)

This notebook trains a DDPM based model on MNIST digits dataset.

python

!pip install labml-nn --quiet

python

from labml import experiment
from labml_nn.diffusion.ddpm.experiment import Configs

python

experiment.create(name="diffuse", writers={'screen'})

python

configs = Configs()

Set experiment configurations and assign a configurations dictionary to override configurations

python

experiment.configs(configs, {
    'dataset': 'MNIST',
    'image_channels': 1,
    'epochs': 5,
})

Initializ

python

configs.init()

Set PyTorch models for loading and saving

python

experiment.add_pytorch_models({'eps_model': configs.eps_model})

python

# Start the experiment
with experiment.start():
    configs.run()