简易去噪扩散模型:一个PyTorch实现
2025-04-03
这是一个用PyTorch实现的简易去噪扩散模型代码库。作者基于已有的优秀资源,重构代码,使其更易于理解。代码逻辑清晰,将功能函数和类分别放在不同的文件中,方便学习者理解扩散模型的构建模块。模型训练和图像生成过程也分别在不同的文件中实现,方便参数调整。虽然生成的图像质量还有提升空间,但这个项目对于学习扩散模型非常有帮助。
开发