#PyTorch Lightning
pytorch-forecasting - 前沿的时间序列预测工具包,提供灵活的高层API
Github开源项目深度学习神经网络PyTorch LightningPyTorch Forecasting时间序列预测
PyTorch Forecasting 是一个基于 PyTorch 的时间序列预测包,适用于实际应用和研究。它支持多种神经网络架构及自动日志记录,利用 PyTorch Lightning 实现多 GPU/CPU 的扩展训练,并内置模型解释功能。关键特性包括时间序列数据集类、基本模型类、增强的神经网络架构、多视角时间序列指标和超参数优化。安装简便,支持 pip 和 conda,文档详尽,并包含模型比较和使用案例。
lightning-bolts - PyTorch Lightning的深度学习组件扩展
Github开源项目PyTorch LightningBoltsTorch ORTSparseMLDeepSparse
Lightning Bolts为PyTorch Lightning提供了多种扩展组件,包括回调和数据集,旨在加速训练和推理。它支持通过Torch ORT将模型转换为优化的ONNX图,以实现GPU加速训练;并通过SparseML在微调中引入稀疏性,提高推理性能。项目支持广泛的问题解决,并欢迎用户贡献通用组件。了解更多安装和使用信息,请访问官方文档和社区支持平台。
pytorch-lightning - 深度学习框架的全方位AI模型训练与部署解决方案
Github开源项目深度学习PyTorch LightningAI模型训练Lightning Fabric模型部署热门
深度学习框架Pytorch-Lightning 2.0版本现已推出,提供清晰稳定的API,支持AI模型的预训练、微调和部署。该框架轻松实现Pytorch代码组织,将科学研究与工程实现分离,帮助研究人员和工程师高效进行模型训练与部署。通过提供各种训练和部署选项以及兼容多种硬件和加速器,Pytorch-Lightning兼顾模型的灵活性和可扩展性,适应从初学者到专业AI研究的不同需求。
lightning-hydra-template - 支持多GPU和TPU等多种训练选项的深度学习模板
Github开源项目深度学习PyTorch Lightning配置管理AI研究Hydra
了解Lightning-Hydra-Template,这是一个兼具代码整洁和高性能的深度学习项目模板。它利用PyTorch Lightning和Hydra优化项目架构和实验管理,支持多GPU和TPU等多种训练选项,同时提供自动化测试和代码风格指导,帮助AI专业人员提升工作效率。此模板适用于快速试验和研发创新。
foldingdiff - Protein生成和优化的扩散模型工具
Github开源项目PyTorch Lightning扩散模型foldingdiff蛋白质生成CATH数据集
使用扩散模型生成新的蛋白骨架结构。提供详细的安装、数据下载和模型训练指南,支持预训练模型的加载和采样。评估生成骨架的设计性能和结构一致性,适合具备Python和PyTorch基础的研发人员使用,通过浏览器便捷地生成蛋白质结构。
solo-learn - 使用自监督学习进行无监督视觉表征的方法与技巧
Github开源项目PyTorch Lightning自监督学习solo-learn视觉表示学习无监督
solo-learn库基于PyTorch Lightning,提供多种自监督方法用于无监督视觉表征学习。该库包含全面的训练技巧和多种数据处理、评估方式,以提高训练效果和可重复性。其主要特点有快速的数据处理、自定义模型检查点、线上和线下的K-NN评估。库内包含灵活的数据增强、可视化功能,并不断更新方法和改进教程,使模型训练和调试更加高效简便。
pytorch_tabular - 表格数据深度学习的简易化解决方案
Github开源项目PyTorch深度学习PyTorch Lightning模型定制标签数据
PyTorch Tabular是一个开源库,旨在简化表格数据的深度学习应用。该库具有低阻力易用性、易于定制和可扩展性,基于PyTorch和PyTorch Lightning构建。提供如TabNet、NODE和GATE等多种模型选择,适用于研究和实际应用。详细的文档和简单的安装流程使用户能够快速上手和自定义模型,提高机器学习任务的效率和性能。
yet-another-lightning-hydra-template - 深度学习项目模板整合PyTorch Lightning和Hydra提升效率
Github开源项目深度学习机器学习PyTorch LightningHydra可复现性
该项目模板基于PyTorch Lightning和Hydra,旨在提高深度学习工作流效率和实验可重复性。模板结构模块化且可扩展,适用于分类、分割和度量学习等任务,并可扩展至其他领域。集成最佳实践和详细文档,支持在多种硬件上进行实验,便于快速迭代和比较不同模型方法。
skillful_nowcasting - DGMR模型,革新短期天气预报技术
Github开源项目PyTorch LightningHuggingFace深度生成模型DGMR雷达预报
本项目是DeepMind的Skillful Nowcasting GAN深度生成模型(DGMR)的开源实现,专注于提高短期天气预报精度。基于PyTorch Lightning框架开发,严格遵循DeepMind公布的伪代码。项目集成了预训练模型,支持英国和美国的降水雷达数据,并通过HuggingFace Datasets简化了数据获取流程。DGMR模型展示了生成高质量短期天气预报的能力,为气象预报领域带来了创新。
bert-base-uncased-emotion - BERT模型用于情感分析的优化与应用
Github开源项目PyTorch Lightning模型数据集情感分析Huggingfacebert-base-uncased-emotion情感类别
该项目基于bert-base-uncased模型,并使用PyTorch Lightning技术在一个情感数据集上进行了微调,支持文本分类和情感分析。训练参数包括128的序列长度、2e-5的学习率、32的批处理大小和4个训练周期,运行在两块GPU上。尽管模型尚未最优化,但在实际应用中显示出一定效果,达到了0.931的验证精度。更多项目详情可以通过nlp viewer查看。
相关文章
PyTorch Forecasting: 简化神经网络时间序列预测
3 个月前
Lightning Bolts: 深度学习研究的强大工具箱
3 个月前
Lightning-Hydra-Template: 一个用户友好的机器学习实验模板
3 个月前
solo-learn: 一个强大的自监督视觉表示学习库
3 个月前
PyTorch Tabular: 一个用于表格数据深度学习的强大框架
3 个月前
Yet Another Lightning Hydra Template: 提升机器学习项目的效率与可复现性
3 个月前
熟练的深度生成模型降水临近预报
3 个月前
pytorch-forecasting学习资料汇总 - 基于PyTorch的高级时间序列预测库
2 个月前
Lightning Bolts 入门学习资料 - AI/ML研究者的模型、回调和数据集工具箱
2 个月前