fastai

[source]

learner

icevision.models.torchvision.mask_rcnn.fastai.learner(dls, model, cbs=None, **learner_kwargs)

Fastai Learner adapted for Mask RCNN.

Arguments

• dls List[Union[torch.utils.data.dataloader.DataLoader, fastai.data.load.DataLoader]]: Sequence of DataLoaders passed to the Learner. The first one will be used for training and the second for validation.
• model torch.nn.modules.module.Module: The model to train.
• cbs: Optional Sequence of callbacks.
• **learner_kwargs: Keyword arguments that will be internally passed to Learner.

Returns

A fastai Learner.

---