电脑基础 · 2023年4月19日

DDPM代码详细解读(1):数据集准备、超参数设置、loss设计、关键参数计算

Diffusion Models专栏文章汇总:入门与实战

前言:大部分DDPM相关的论文代码都是基于《Denoising Diffusion Probabilistic Models》和《Diffusion Models Beat GANs on Image Synthesis》贡献代码基础上小改动的。官方的DDPM是tensorflow TPU版本,暂时没有GPU的版本。本篇文章开始,详细解读一下pytorch和tensorflow版本的代码。

目录

超参数设置

训练CIFAR10数据集的配置信息