热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#UNet
unet.cu
这个开源项目使用纯C++/CUDA实现了UNet扩散模型训练框架,支持无条件扩散。框架包含线性层、组归一化、注意力等核心算子的GPU加速实现,重点优化3x3卷积。通过多次迭代提升CUDA kernel性能,训练速度达PyTorch的40%。项目展示了深度学习框架在GPU上的高效实现过程,为相关开发提供参考。
DiffusionFromScratch
DiffusionFromScratch是一个开源项目,提供精简代码库用于重建稳定扩散模型。项目特点包括单Python脚本实现、支持MNIST和CelebA数据集训练,以及提供多个Colab笔记本。这些笔记本涵盖模型架构探索、UNet模型构建和基于文本生成MNIST图像等内容。项目还展示了演示输出和音乐视频生成示例,为学习稳定扩散模型提供了实用资源。
相关文章
UNet.cu: 用纯CUDA实现UNet扩散模型
1 个月前
从零开始理解和实现扩散模型:DiffusionFromScratch 项目详解
1 个月前
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号