Welcome to fastai
fastai simplifies training fast and accurate neural nets using modern best practices
Important: This documentation covers fastai v2, which is a from-scratch rewrite of fastai. The v1 documentation has moved to fastai1.fast.ai. To stop fastai from updating to v2, run in your terminal
echo 'fastai 1.*' >> $CONDA_PREFIX/conda-meta/pinned (if you use conda).
You can use fastai without any installation by using Google Colab. In fact, every page of this documentation is also available as an interactive notebook – click “Open in colab” at the top of any page to open it (be sure to change the Colab runtime to “GPU” to have it run fast!) See the fast.ai documentation on Using Colab for more information.
You can install fastai on your own machines with conda (highly recommended). If you’re using Anaconda then run:
conda install -c fastai -c pytorch -c anaconda fastai gh anaconda
…or if you’re using miniconda) then run:
conda install -c fastai -c pytorch fastai
To install with pip, use:
pip install fastai. If you install with pip, you should install PyTorch first by following the PyTorch installation instructions.
If you plan to develop fastai yourself, or want to be on the cutting edge, you can use an editable install (if you do this, you should also use an editable install of fastcore to go with it.):
git clone https://github.com/fastai/fastai pip install -e "fastai[dev]"
To see what’s possible with fastai, take a look at the Quick Start, which shows how to use around 5 lines of code to build an image classifier, an image segmentation model, a text sentiment model, a recommendation system, and a tabular model. For each of the applications, the code is much the same.
Read through the Tutorials to learn how to train your own models on your own datasets. Use the navigation sidebar to look through the fastai documentation. Every class, function, and method is documented here.
To learn about the design and motivation of the library, read the peer reviewed paper.
fastai is a deep learning library which provides practitioners with high-level components that can quickly and easily provide state-of-the-art results in standard deep learning domains, and provides researchers with low-level components that can be mixed and matched to build new approaches. It aims to do both things without substantial compromises in ease of use, flexibility, or performance. This is possible thanks to a carefully layered architecture, which expresses common underlying patterns of many deep learning and data processing techniques in terms of decoupled abstractions. These abstractions can be expressed concisely and clearly by leveraging the dynamism of the underlying Python language and the flexibility of the PyTorch library. fastai includes:
- A new type dispatch system for Python along with a semantic type hierarchy for tensors
- A GPU-optimized computer vision library which can be extended in pure Python
- An optimizer which refactors out the common functionality of modern optimizers into two basic pieces, allowing optimization algorithms to be implemented in 4–5 lines of code
- A novel 2-way callback system that can access any part of the data, model, or optimizer and change it at any point during training
- A new data block API
- And much more…
fastai is organized around two main design goals: to be approachable and rapidly productive, while also being deeply hackable and configurable. It is built on top of a hierarchy of lower-level APIs which provide composable building blocks. This way, a user wanting to rewrite part of the high-level API or add particular behavior to suit their needs does not have to learn how to use the lowest level.
Migrating from other libraries
It’s very easy to migrate from plain PyTorch, Ignite, or any other PyTorch-based library, or even to use fastai in conjunction with other libraries. Generally, you’ll be able to use all your existing data processing code, but will be able to reduce the amount of code you require for training, and more easily take advantage of modern best practices. Here are migration guides from some popular libraries to help you on your way:
To run the tests in parallel, launch:
For all the tests to pass, you’ll need to install the following optional dependencies:
pip install "sentencepiece<0.1.90" wandb tensorboard albumentations pydicom opencv-python scikit-image pyarrow kornia \ catalyst captum neptune-cli
Tests are written using
nbdev, for example see the documentation for
After you clone this repository, please run
nbdev_install_git_hooks in your terminal. This sets up git hooks, which clean up the notebooks to remove the extraneous stuff stored in the notebooks (e.g. which cells you ran) which causes unnecessary merge conflicts.
Before submitting a PR, check that the local library and notebooks match. The script
nbdev_diff_nbs can let you know if there is a difference between the local library and the notebooks.
- If you made a change to the notebooks in one of the exported cells, you can export it to the library with
- If you made a change to the library, you can export it back to the notebooks with
For those interested in official docker containers for this project, they can be found here.