English
库 / 框架

denoising-diffusion-pytorch

lucidrains/denoising-diffusion-pytorch

用PyTorch实现的去噪扩散概率模型库,提供U模型、扩散过程与训练器,方便训练图像与序列生成模型。

标签