← 返回列表

Tensor2Tensor:面向多模态任务的深度学习模型与数据集库

tensorflow/tensor2tensor · Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

GitHub原文发布 未知首次收录 2026-10-30上架 2026-10-30 07:02

☆ 收藏查看项目 ↗

ELI5:它是什么,解决什么问题

Tensor2Tensor(T2T)由 Google Brain 团队与社区开发,是一个汇集深度学习模型和数据集的 Python 库,覆盖数学语言理解、故事问答、图像分类与生成、语言建模、情感分析、语音识别、摘要和机器翻译等任务。它提供 t2t-datagen 数据生成、t2t-trainer 训练评估推理入口,以及命令行切换问题、模型和超参数集,并支持多 GPU 和分布式训练。项目已标记为弃用,仍维护并可接受缺陷修复,官方建议转向后继库 Trax。

核心能力与快速开始

★ 17473 · Python · 已归档

能做什么
提供内置深度学习模型库与多模态数据集库,统一以 Problem(数据与模态)和 Model(张量到张量计算)组织;通过注册机制和命令行参数在数据生成、训练、解码、评测之间切换,支持 Transformer、Universal Transformer、ResNet、Xception、Shake-Shake、Image Transformer、Transformer AE 等模型及多种超参数集。
适用场景
可用于机器翻译、文本摘要、语言建模、情感分析、语音识别、图像分类与生成、故事问答和数学语言理解等任务的训练与研究实验;也可作为自定义模型、数据集、超参数集和模态的扩展基础。
项目形态
开发框架
部署方式
未说明
主要依赖
需安装 TensorFlow 或 tensorflow-gpu 后通过 pip 安装 tensor2tensor;README 提及通常在 Cloud TPU 或 8-GPU 机器上运行,其他环境可能需要调整超参数。
使用限制
项目已标记为弃用,仅保持运行并欢迎缺陷修复,官方建议使用后继库 Trax;未说明具体 API 成本、硬件门槛或商用限制。
许可证
Apache-2.0
最近推送
2023-06-03 02:55

资料检查:2026-10-30 07:00 · 卡片整理:2026-10-30 07:00
卡片由 AI 根据仓库资料整理,未进行安装或运行验证

职业应用

程序员

作为深度学习模型与数据集的 Python 库,程序员可用 t2t-datagen 生成数据、用 t2t-trainer 训练/评估/推理,并通过命令行切换问题、模型和超参数集;支持多 GPU 和分布式训练,可用于机器翻译、摘要、语音识别等任务的模型开发与实验。

查看内容依据
它提供 t2t-datagen 数据生成、t2t-trainer 训练评估推理入口,以及命令行切换问题、模型和超参数集,并支持多 GPU 和分布式训练。

官方使用入口 ↗

由 AI 根据来源材料整理,未进行安装或运行验证

这是基于来源内容整理的摘要。原文链接失效后,本页摘要和讨论仍会保留。

讨论 0

登录后参与讨论,阅读资讯无需注册。

还没有评论,来分享第一个看法。