1
Aug
2019
/
Boris Dayma, ML Engineer

Visualize Fastai Faster

If you're using fastai, it's now easier than ever to log, visualize, and compare your experiments. Just import wandb and add our callback:

import wandb
from wandb.fastai import WandbCallback
wandb.init()
learn = cnn_learner(data, model, callback_fns=WandbCallback)

Learn more in the docs →

What does the integration get you?

Add wandb, and you'll get a powerful, persistent, and shareable dashboard for exploring your results and comparing experiments. Here are a few snapshots from a project where I'm comparing the ground truth and predictions in my semantic segmentation project.

I'm able to look at example outputs, visually compare versions of my model, and identify anomalies.

Here are some graphs from the same fastai project. I like this as an alternative to TensorBoard or TensorBoardX because W&B keeps the hyperparameters, metric graphs, and checkpointed model versions organized automatically. I can send a link to share my findings, and collaborators can explore my results independently without relying on my screenshots of local TensorBoard instances. It's also nice to know that the results are always saved in the cloud, so I never have to dig through messy local files.

If you'd like to try the fastai integration on a quick example problem, clone my image classification repo and try your hand at classifying Simpsons characters. Give it a try →

Newsletter

Enter your email to get updates about new features and blog posts.

Weights & Biases

We're building lightweight, flexible experiment tracking tools for deep learning. Add a couple of lines to your python script, and we'll keep track of your hyperparameters and output metrics, making it easy to compare runs and see the whole history of your progress. Think of us like GitHub for deep learning.

Partner Program

We are building our library of deep learning articles, and we're delighted to feature the work of community members. Contact Carey to learn about opportunities to share your research and insights.

Try our free tools for experiment tracking →