PyTorch笔记 - Diffusion Model 源码开发
Posted SpikeKing
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch笔记 - Diffusion Model 源码开发相关的知识,希望对你有一定的参考价值。
Diffusion Model的效果如下:
源码如下:
- 选择一个数据集
- 确定超参数的值
- 确定扩散过程任意时刻的采样值
- 演示原始数据分布加噪100步后的效果
- 编写拟合逆扩散过程高斯分布的模型
- 编写训练的误差函数
- 编写逆扩散采样函数(inference过程)
- 开始训练模型,并打印loss及中间的重构效果
- 动画演示扩散过程和逆扩散过程
# 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)