PyTorch笔记 - Diffusion Model 源码开发

Posted 2023-03-03 SpikeKing

Diffusion Model的效果如下：

源码如下：

1. 选择一个数据集
2. 确定超参数的值
3. 确定扩散过程任意时刻的采样值
4. 演示原始数据分布加噪100步后的效果
5. 编写拟合逆扩散过程高斯分布的模型
6. 编写训练的误差函数
7. 编写逆扩散采样函数(inference过程)
8. 开始训练模型，并打印loss及中间的重构效果
9. 动画演示扩散过程和逆扩散过程

# 1、选择一个数据集
%matplotlib inline

import matplotlib.pyplot as plt
import numpy as np
from sklearn.datasets import make_s_curve
import torch

s_curve