58、Improved Diffusion的PyTorch代码逐行深入讲解

Improved Diffusion

IDDPM

diffusion 相比gan优势

attention unet

字典扩充

add_dict_to_argparse

自动化定义

从字典中自动生成argument parser 很方便，不用一个个手打

Load data

list_image_files_recursively(data_dir)循环递归找图片

ImageDataset

trainloop

create_model_and_diffusion

高斯diffusion加噪法

space diffusion

高斯diffusion