PyTorch笔记 - Diffusion Model 源码开发

Posted SpikeKing

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch笔记 - Diffusion Model 源码开发相关的知识,希望对你有一定的参考价值。

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笔记 - Diffusion Model 源码开发的主要内容,如果未能解决你的问题,请参考以下文章