SCENE COLLECTION

模型训练 方向开源项目精选

这里汇总了代码库收录的「模型训练」场景开源项目,按 Star 总量排序,适合按方向系统性地挑选与对比工具。

共收录 83 个项目 · 第 4 / 5 页

61

lectures

oxford-cs-deepnlp-2017 / lectures

牛津大学2017年深度自然语言处理课程,包含全套讲义与视频,系统讲解NLP核心技术和深度学习方法。

编辑点评经典NLP课程,内容覆盖RNN、LSTM、注意力机制、机器翻译等关键主题,适合想扎实掌握深度学习NLP基础的开发者。缺点是年份较早,未涉及Transformer、预训练大模型等最新进展,但核心原理讲解清晰,搭配作业和代码实践仍有很高学习价值。

数据科学 模型训练 深度学习 自然语言处理 课程资料 牛津大学 NLP
最近提交 2023-07-02
16k 3.5k ▲ 1
62

cheatsheets-ai

kailashahirwar / cheatsheets-ai

面向机器学习与深度学习研究者的速查表合集,汇总常用框架、算法和数学知识,帮助快速查阅核心概念与代码。

编辑点评这个项目把散落的各类 ML/DL 速查表集中整理,覆盖 TensorFlow、PyTorch、NumPy、Pandas 等常用工具,适合初学者快速上手和研究者随时翻查。缺点是部分内容可能更新不及时,且缺少统一维护,建议结合官方文档使用。非常适合作为学习笔记和日常开发的便捷参考。

数据科学 开发工具 模型训练 机器学习 深度学习 速查表 神经网络 数据科学
MIT · 最近提交 2019-10-19
15k 3.4k ▲ 1
63

albumentations

albumentations-team / albumentations

Albumentations 是一个快速灵活的 Python 图像增强库,为深度学习模型训练提供丰富的数据增强操作,提升模型泛化能力。

编辑点评它是计算机视觉领域最流行的数据增强工具之一,基于 OpenCV 高效实现,支持多种主流深度学习框架。与 PyTorch、TensorFlow 无缝集成,适合图像分类、检测、分割等任务。丰富的增强策略和简洁的 API 让开发者能快速构建数据流水线,但在超大批量场景下需注意 CPU 瓶颈。

模型训练 数据科学 AI 基础设施 图像增强 数据增强 深度学习 计算机视觉 Python
MIT · 最近提交 2025-06-25
Python 15k 1.7k ▲ -1
64

DeepLearningExamples

NVIDIA / DeepLearningExamples

NVIDIA官方维护的深度学习示例集合,按模型分类,提供可复现的训练与部署脚本。

编辑点评这是NVIDIA精心整理的示例库,覆盖图像、语音、推荐等主流模型,代码质量高,针对GPU训练做了深度优化,适合在NVIDIA企业级基础设施上快速复现SOTA结果。优点是权威性和工程化程度高,示例完整;但资源消耗大,且强依赖NVIDIA硬件,入门门槛较高。适合已有一定深度学习基础、希望参考工业级实现或进行性能调优的开发者。

模型训练 AI 基础设施 数据科学 深度学习 NVIDIA 模型训练 Jupyter Notebook 示例代码
最近提交 2024-08-12
Jupyter Notebook 15k 3.4k ▲ 1
65

imgaug

aleju / imgaug

imgaug 是一个用于机器学习实验的 Python 图像增强库,提供丰富多样的数据增强操作,帮助提升模型泛化能力。

编辑点评imgaug 是图像增强领域的老牌经典库,支持几何变换、颜色扰动、噪声添加等数十种增强策略,且可组合使用。适合需要快速扩充图像数据集、训练鲁棒视觉模型的场景。优点是功能全面、接口灵活、文档详尽;缺点是维护活跃度下降,新项目可考虑与 albumentations 对比选择,但已有代码迁移成本低。

数据科学 模型训练 AI 基础设施 图像增强 数据增强 机器学习 Python
MIT · 最近提交 2024-07-30
Python 15k 2.4k ▲ 0
66

horovod

horovod / horovod

面向 TensorFlow、Keras、PyTorch 和 Apache MXNet 的分布式训练框架,帮助用户将单机训练脚本无缝扩展到多机多卡环境。

编辑点评Horovod 以高性能环形全归约算法著称,能显著降低分布式训练通信开销,且对现有代码侵入小,是传统深度学习场景中成熟的分布式方案。它适合已有单机训练代码、需要快速扩展至多机多卡的研究团队或生产环境。注意其设计偏重同步训练,对大规模异步或动态稀疏训练支持有限,且社区活跃度较早期有所下降,新项目可评估更现代的框架。

模型训练 AI 基础设施 分布式训练 TensorFlow PyTorch Keras MXNet
NOASSERTION · 最近提交 2026-09-17
Python 15k 2.2k ▲ 0
67

transferlearning

jindongwang / transferlearning

迁移学习/领域自适应/领域泛化/多任务学习等方向的论文、代码、数据集、应用和教程汇总,是系统学习和查阅该领域资源的必备仓库。

编辑点评内容覆盖面广,从经典方法到前沿方向都有整理,并附有代码和数据集链接,非常适合科研入门和工程选型参考。仓库以资源索引为主,需要自己动手阅读和实验;部分外部链接可能失效,建议结合论文原文核对。总体是迁移学习领域不可多得的导航型项目。

数据科学 模型训练 AI 基础设施 迁移学习 领域自适应 领域泛化 多任务学习 学习资源
MIT · 最近提交 2025-02-18
Python 14k 3.8k ▲ 1
68

nni

microsoft / nni

微软开源的 AutoML 工具包,覆盖特征工程、神经架构搜索、模型压缩与超参调优,自动化机器学习全流程。

编辑点评NNI 是微软出品的成熟 AutoML 框架,提供统一 API 和可视化界面,适合需要系统化调参、自动搜索网络结构或压缩模型的团队。其优势在于模块丰富、支持分布式训练,但学习曲线较陡,配置项多,社区更新频率近年有所下降。若想快速跑通实验,建议从内置示例开始。

模型训练 AI 基础设施 数据科学 AutoML 神经架构搜索 超参调优 模型压缩 特征工程
MIT · 最近提交 2024-07-03
Python 14k 1.9k ▲ 0
69

mit-deep-learning-book-pdf

janishar / mit-deep-learning-book-pdf

MIT 深度学习书籍的 PDF 版本(完整版与分章节版),涵盖 Ian Goodfellow、Yoshua Bengio 和 Aaron Courville 的经典教材内容,方便离线阅读与查阅。

编辑点评这是深度学习领域公认的必读经典,原著内容系统全面,覆盖数学基础、深度网络、优化、正则化及前沿方向。此仓库提供便捷的 PDF 格式,适合希望在无网络环境下学习或作为案头参考的开发者。注意其语言标记为 Java 仅是仓库默认代码语言,与内容无关;建议结合官方英文原版或中文译本使用,以获取最新勘误。

数据科学 模型训练 AI 基础设施 深度学习 经典教材 PDF Ian Goodfellow 机器学习
最近提交 2026-02-03
Java 14k 2.9k ▲ 0
70

litgpt

Lightning-AI / litgpt

一个基于 PyTorch 的开源大模型工具包,提供 20+ 高性能 LLM 的预训练、微调和规模化部署方案,让开发者更高效地落地大模型。

编辑点评LitGPT 将 Lightning 生态的工程经验带入 LLM 全流程,配置简洁、脚本可控,适合研究团队或中小公司快速复现主流模型。它支持从零预训练、LoRA/QLoRA 微调及推理部署,但依赖 transformers 生态相对较深,需要一定 PyTorch 基础。若你想避开重量级框架,追求灵活性和透明性,它是不错的选择。

模型训练 AI 基础设施 LLM 应用 大语言模型 预训练 微调 PyTorch 部署
Apache-2.0 · 最近提交 2026-11-01
Python 14k 1.5k ▲ 0
71

Ai-Learn

tangyudi / Ai-Learn

一份系统的 AI 学习路线图,整合近 200 个实战案例与项目,覆盖 Python、机器学习、深度学习、NLP 等热门领域,适合零基础入门和就业实战。

编辑点评内容覆盖面广,从 Python、数学基础到机器学习、深度学习,再到 CV/NLP 等方向均有案例支撑,免费配套教材降低了学习门槛。适合想系统入门 AI 或准备就业的开发者。但路线图类资源需要自行把握学习节奏,部分案例可能偏旧,建议结合最新框架版本实践。

数据科学 模型训练 开发工具 AI学习路线 机器学习 深度学习 实战项目 数据分析
最近提交 2024-06-02
13k 2.7k ▲ 2
72

pytorch-grad-cam

jacobgil / pytorch-grad-cam

PyTorch 实现的 Grad-CAM 可解释性工具,支持 CNN、Vision Transformer 及分类、检测、分割等任务的视觉解释。

编辑点评这是计算机视觉可解释性领域最流行的库之一,API 简洁且支持多种模型架构和任务类型,能快速生成热力图辅助调试模型。适合算法工程师分析模型关注区域、验证鲁棒性,也适合论文实验可视化。注意不同版本的 PyTorch 和模型结构可能需要适配,且可视化结果需结合实际任务解读。

数据科学 开发工具 模型训练 可解释性 Grad-CAM 计算机视觉 PyTorch 可视化
MIT · 最近提交 2026-11-01
Python 13k 1.7k ▲ 0
73

deep-learning-drizzle

kmario23 / deep-learning-drizzle

汇集深度学习、强化学习、机器学习、计算机视觉和NLP领域的顶尖公开课视频,助你通过系统学习掌握AI核心知识。

编辑点评这是一份精心整理的AI学习资源清单,覆盖多个关键方向,并附有分类和课程链接,适合想系统入门或进阶的开发者。优点是资源丰富且免费,按主题和时间组织,方便查找;缺点是部分视频为外部托管,可能需要自行处理访问问题。建议配合课程资料动手实践,效果更佳。

数据科学 模型训练 深度学习 强化学习 机器学习 计算机视觉 自然语言处理
最近提交 2026-08-22
HTML 13k 3k ▲ 0
74

fashion-mnist

zalandoresearch / fashion-mnist

一个替代 MNIST 的时尚产品图像数据集,包含 10 类服装灰度图,用于评估机器学习模型的基准测试。

编辑点评Fashion-MNIST 是 MNIST 的现代替代品,图像更复杂、分类更具挑战性,适合检验模型在简单图像任务上的真实表现。数据规模与 MNIST 一致(7 万张 28×28 图像),可直接替换使用,但精度天花板较低,更适合入门教学和快速验证,而非前沿研究。注意其类别不平衡问题较小,但图像仍为灰度,不适合直接迁移到彩色图像任务。

数据科学 模型训练 图像分类 基准数据集 机器学习 TensorFlow PyTorch
MIT · 最近提交 2022-06-13
Python 13k 3.1k ▲ 2
75

tensorzero

tensorzero / tensorzero

Rust 平台,涉及AI、AI-engineering、anthropic

编辑点评适用于AI/大模型、基础设施场景。使用 Rust 开发,涉及AI、AI-engineering、anthropic等技术方向。建议关注其社区活跃度和文档完善程度。

LLM 应用 模型训练 Rust AI AI-engineering anthropic artificial-intelligence deep-learning
Apache-2.0 · 最近提交 2026-06-11
Rust 12k 967 ▲ 1
76

maths-cs-ai-compendium

HenryNdubuaku / maths-cs-ai-compendium

TypeScript 引擎,涉及AI-textbook、algorithms、artificial-intelligence

编辑点评适用于AI/大模型、基础设施、数据科学场景。使用 TypeScript 开发,涉及AI-textbook、algorithms、artificial-intelligence等技术方向。建议关注其社区活跃度和文档完善程度。

模型训练 数据科学 TypeScript AI-textbook algorithms artificial-intelligence computer-science computer-vision
Apache-2.0 · 最近提交 2026-07-18
TypeScript 7.7k 945 ▲ 3
77

models

PaddlePaddle / models

飞桨官方模型库,涵盖计算机视觉、自然语言处理、语音、推荐、时序预测及大模型等领域的经典与前沿模型实现。

编辑点评这是 PaddlePaddle 官方的模型集合,覆盖面广且持续维护,适合需要基于飞桨快速搭建基线模型的开发者。相比其他模型库,它更贴近国产深度学习生态,中文文档和社区支持较好。不过部分模型更新可能滞后,需注意与当前 PaddlePaddle 版本的兼容性。

模型训练 AI 基础设施 数据科学 飞桨 模型库 深度学习 CV NLP
Apache-2.0 · 最近提交 2025-01-15
Python 6.9k 2.8k ▲ 0
78

donut

clovaai / donut

Donut是OCR-free文档理解Transformer,无需OCR即可直接解析文档图像,并配套合成文档生成器SynthDoG。

编辑点评创新性地跳过OCR环节,避免误差累积,直接端到端理解文档结构,适合票据、表单等结构化信息抽取场景。训练需较多算力,且合成数据与真实数据存在分布差异,建议配合领域数据微调。

数据科学 模型训练 文档理解 OCR-free Transformer 合成数据 信息抽取
MIT · 最近提交 2024-07-11
Python 6.9k 562 ▲ 0
79

courses

SkalskiP / courses

一个精心整理的 AI 课程与学习资源合集,汇集了关于人工智能的各类教程、课程和资料链接,帮助开发者系统学习 AI 知识。

编辑点评该仓库由 PyTorch 知名贡献者维护,链接质量高且覆盖面广,从机器学习基础到深度学习的进阶内容都有涉及。适合刚入门 AI 的开发者按图索骥,也适合有经验者查漏补缺。不过它只是资源导航,不包含原创教程,实际学习需要自行访问外部链接,部分内容可能需科学上网。

数据科学 模型训练 开发工具 AI课程 学习资源 人工智能 精选列表 GitHub
最近提交 2024-04-22
Python 6.5k 595 ▲ 0
80

TensorFlow-2.x-Tutorials

dragen1860 / TensorFlow-2.x-Tutorials

TensorFlow 2.x 入门实战教程与示例代码,涵盖 CNN、RNN、GAN、BERT 等经典模型,帮助开发者快速上手深度学习。

编辑点评这是 TF 2.0 时代广受欢迎的教程仓库,示例丰富且代码可直接运行,覆盖从基础模型到前沿架构。适合想系统学习 TensorFlow 2 的初学者,也适合需要参考实现的研究者。注意部分示例基于 TF 2.x 早期版本,新版本 API 可能需要微调。

模型训练 数据科学 开发工具 TensorFlow 2.x 深度学习教程 实战示例 神经网络 BERT/GPT
最近提交 2020-09-23
Jupyter Notebook 6.3k 2.2k ▲ 0

相关场景

继续浏览