transformers

用于PyTorch、TensorFlow和JAX的最先进的机器学习库。

Python
Hugging Face
机器学习
初始发布时间:2018年12月
工具描述

transformers是Hugging Face发布并维护的一个机器学习库,它提供了数以千计的预训练模型,用以处理各种任务,包括文本、视觉和音频等。

最开始,Hugging Face发布的并不是transformers,而是著名的pytorch_pretrained_bert。也就是bert刚出来之后不久,他们实现了一个基于PyTorch的bert模型,这一下就火了起来。于是,Hugging Face也抓住了这个机会,不断地对它进行改进。并趁着预训练模型的发展不断的增加transformers的能力,以至于现在几乎是搞transformers模型都在使用的开源库。

transformers可以实现如下任务:

  • 📝文本:文本分类、信息提取、问题回答、总结、翻译和100多种语言的文本生成。
  • 🖼️ 图像:图像分类、对象检测和分割。
  • 🗣️ 音频:语音识别和音频分类。
  • 🐙多模态:表格问题回答、光学字符识别、扫描文件的信息提取、视频分类和视觉问题回答。

是否开源:

许可协议: Apache-2.0 license

官方地址: https://huggingface.co/docs/transformers/index

GitHub地址: https://github.com/huggingface/transformers

初始贡献者: Thomas Wolf等

最佳实践指南

官方指南:https://huggingface.co/docs/transformers/index

Hugging Face-logo
pytorch-logo
推荐工具

TensorFlow - 深度学习

MindSpore - 深度学习