学习PyTorch 写一个极简 DDPM(Diffusion Model),理解图片生成模型到底怎么训练和生成
這條路和現在主流圖片生成模型的核心思想是相通的:訓練時把圖片逐步加噪,讓神經網路學會預測噪聲;生成時從純噪聲開始,一步一步把噪聲去掉。 Hugging Face 現在的官方訓練範例也是用 UNet + DDPM scheduler 做這件事。一、我們先做一個「真的能跑」的最小模型我建議第一版不要做: