【源码解读】扩散模型核心：DDPM专题-结合源码讲解

本次训练采用的是cifar数据集，代码和下载好的数据将打包上传在百度网盘。

1. 训练

1.1 Uniform({1,…,T})

训练过程, t是随机采样获得的, 这一步是核心之一, 相当于伪代码中的 Step3: t ∼ Uniform ⁡ ( { 1 , … , T } ) t \sim \operatorname{Uniform}(\{1, \ldots, T\}) t∼Uniform({ 1,…,T})

1.2 $\mathbf{ϵ}\sim \mathcal{N}(0,\mathbf{I})$

代码142行：生成均值为1，方差为0的标准高斯分布噪声

注意一个细节，t的维度是128，表示一个batchsize一起进行加噪

1.3 加噪

主要的函数代码在144行，这里将随机采样的加噪时间t，生成的noise和一个bath的image一起放入perturb函数中。

这里的加噪公式对应论文中的：$q($