fastai/fastai

The fastai deep learning library

What it solves

fastai 是一个深度学习库,旨在让标准深度学习领域的最新成果变得易于获取且容易实现。它弥合了实践者需要快速生产力的高层组件与研究者需要灵活可 hack 的低层组件之间的鸿沟。

How it works

基于 PyTorch 构建,fastai 使用分层的解耦抽象架构。这使得用户可以选择使用的 API 层级——从用于图像分类或文本情感分析等标准任务的高层简洁代码,到用于自定义研究的低层构建块。关键技术特性包括 GPU 优化的计算机视觉库、创新的双向回调系统用于训练修改、新的数据块 API,以及自定义的 Python 类型分派系统。

Who it’s for

它面向既想快速构建和部署模型的深度学习实践者,也面向需要高度可配置框架来实验新深度学习技术的研究者。

Highlights

  • 拥有平衡易用性与深度可 hack 的分层 API
  • 支持图像分类、图像分割、文本情感、推荐系统和表格模型
  • 提供 GPU 优化的计算机视觉库
  • 具备灵活的双向回调系统,可在训练期间修改数据、模型或 optimizer
  • 与 PyTorch 及其他基于 PyTorch 的库无缝集成