Note: this documentation is for fastai v1.If you use the current version of fastai, you should refer to fastai page.
For scripts using fastai v1, we have a callback that can automatically log model topology, losses, metrics, weights, gradients, sample predictions and best trained model.
1
import wandb
2
from wandb.fastai import WandbCallback
3
4
wandb.init()
5
6
learn = cnn_learner(data,
7
model,
8
callback_fns=WandbCallback)
9
learn.fit(epochs)
Copied!
Requested logged data is configurable through the callback constructor.