PyTorch笔记 - Diffusion Model 源码开发 (2)

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

你可能感兴趣的:(深度学习,pytorch,python)