扩散模型实战：从零开始训练手写数字生成模型

Diffusion models代码解读：入门与实战

前言：手写数字的数据集是绝大部分炼丹师的深度学习初恋，这篇博客以代码为主，手把手带读者搭建一个基于扩散模型的手写数字生成模型，非常适合刚接触扩散模型的初学者学习。

目录

环境准备

数据集准备

加噪过程

模型

模型训练

推理

---

环境准备

pip install -q diffusers

import torch
import torchvision
from torch import nn
from torch.nn import functional as F
from torch.utils.data import DataLoader
from diffusers