
denoising_diffusion
用PyTorch Lightning复现DDPM,快速跑通扩散模型训练与生成
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 91
核心亮点
- 代码结构清晰,基于PyTorch Lightning简化训练流程,易于上手
- 完整实现DDPM前向加噪和反向去噪过程,适合学习原理
- 支持自定义数据集和图像尺寸,灵活适配不同实验需求
不足之处
- 项目为早期版本,文档和示例较少,社区支持待观察
- 未提供预训练模型,需自行训练才能评估生成效果
适用场景
- 学习扩散模型原理和PyTorch实现细节
- 快速搭建扩散模型训练实验基线
- 在自定义图像数据集上验证DDPM生成效果
替代项目
openai/guided-diffusion、lucidrains/denoising-diffusion-pytorch、huggingface/diffusers
同类项目推荐
oumi
开源
一条命令搞定 LLM 微调、评估与部署,让模型落地更简单。
Easily fine-tune (SFT/RL), evaluate, and deploy Qwen, Gemma, or any open agentic LLM···
data-forcing-distillation
开源
少步生成视频,不丢多样性和画质
Data-Forcing Distillation (DFD): restoring diversity and fidelity in few-step video ···
MyLLM
开源
跟着 Notebook 从数据到应用跑通大模型全流程
"LLM from Zero to Hero: An End-to-End Large Language Model Journey from Data to Appl···
Visual-ERM
开源
训练视觉奖励模型,让AI输出更符合人类视觉偏好
[NeurIPS 2026] Official Implementation of "Visual-ERM: Reward Modeling for Visual Eq···
CUDA-L2
开源
用强化学习自动写出比 cuBLAS 更快的矩阵乘法内核
CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforceme···
ringattention
开源
环形分布注意力,让长上下文处理又快又省显存
Large Context Attention
ETO
开源
让 LLM Agent 在试错中自我进化,越用越聪明
Trial and Error: Exploration-Based Trajectory Optimization of LLM Agents (ACL 2024 M···
amazon-sagemaker-examples
开源
照着跑一遍,轻松上手 AWS 机器学习全流程
Example Jupyter notebooks that demonstrate how to build, train, and deploy machine ···
Adam-mini
开源
用更少学习率,训更快更大模型,省显存提性能。
Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406···