DDPM代码详细解读(3):图解模型各部分结构、用ConvNextBlock代替Resnet

Diffusion Models专栏文章汇总:入门与实战

前言:前两篇文章讲了pytorch版本的代码,并一一介绍重要方法的原理。官方给的代码是tensorflow TPU版本,没有跑通。但是有很多人用pytorch复现了tensorflow TPU版本的代码,暂时没看见tensorflow GPU版本的代码。这篇文章解读一下对应的pytorch版本的代码,重点分析一下用最新的ConvNextBlock代替ResBlock效果。

目录

整体Unet结构

基于ResBloc

你可能感兴趣的:(Diffusion,Models与深度学习,DDPM,深度学习,tensorflow,python)