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

PyTorch笔记 - Diffusion Model 源码开发

Diffusion扩散模型学习1——Pytorch搭建DDPM实现图片生成

PyTorch - Diffusion Model 公式推导

首个中文Stable Diffusion模型开源;TPU演进十年;18个PyTorch性能优化技巧 | AI系统前沿动态...

[论文笔记]DiffusionCLIP: Text-Guided Diffusion Models for Robust Image Manipulation

论文笔记:DCRNN (Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting)