pytorch-lightning
本站收录 14 个带「pytorch-lightning」标签的 AI 开源项目
Dreambooth-Stable-DiffusionDreambooth-Stable-Diffusion 是 Dreambooth 论文的 Stable Diffusion 实现，旨在解决文本到图像模型中特定主★ 7,731
SUPIRSUPIR 是一个面向真实世界图像修复的开源项目，旨在通过深度学习算法将低质量、模糊或有噪声的照片恢复为高分辨率、逼真的图像。它基于 Stable Diffus★ 5,673
lightning-hydra-template这是一个面向机器学习实验的工程化模板，基于 PyTorch Lightning 和 Hydra 构建。它解决了深度学习项目中配置管理混乱、代码结构不清晰、实验难★ 5,356
TexoTexo 是一个超轻量的 LaTeX 公式识别模型，参数量仅 20M，却能达到 SOTA 级别的识别精度。它主要解决数学公式 OCR 场景中模型过大、难以在浏览★ 903
quaterionQuaterion 是一个专注于相似度学习模型微调的 Python 框架，核心解决的是在检索、去重、聚类等场景下，如何高效训练出能输出高质量向量表示的模型。它把★ 663
open-genieopen-genie 是 DeepMind 论文《Genie: Generative Interactive Environments》的非官方 PyTorch★ 295
yet-another-lightning-hydra-template这是一个基于 PyTorch Lightning 和 Hydra 的机器学习项目模板，旨在为快速实验提供灵活、可扩展且可复现的工作流。它解决了研究人员在搭建新项★ 251
consistency-modelsconsistency-models 是 OpenAI 官方发布的用于训练和采样一致性模型的工具包。一致性模型是一种新型的生成模型，旨在解决传统扩散模型采样速度★ 207
tracemlTraceML 是一个开源的 PyTorch 训练性能诊断工具，旨在帮助开发者定位和解决训练过程中的性能瓶颈。它通过自动捕获和分析训练运行时的性能数据，提供可视★ 185
Neural-HMMNeural-HMM 是一个基于神经隐马尔可夫模型（Neural HMM）的高质量文本转语音（TTS）开源项目。它旨在解决传统注意力机制在语音合成中的鲁棒性不足★ 164
FridoFrido 是一个基于特征金字塔扩散模型的图像生成研究项目，源自论文《Frido: Feature Pyramid Diffusion for Complex ★ 114
denoising_diffusion这是一个非官方的PyTorch实现，复现了论文《Denoising Diffusion Probabilistic Models》（DDPM）中的去噪扩散概率模★ 91
alias-free-ganAlias-Free GAN 是英伟达 StyleGAN3 的非官方实现，基于 rosinality 的版本扩展了训练和推理选项。它解决了传统 GAN 在生成图★ 76
Image-Generation-models这是一个用 PyTorch、PyTorch Lightning 和 Hydra 实现的生成模型集合，涵盖 GAN、VAE、扩散模型和自回归模型四大类。项目旨在为★ 61