Skip to content

fastai

[source]

learner

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

Fastai Learner adapted for Faster RCNN.

Arguments

  • dls Sequence[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[fastai.callback.core.Callback]]: Optional Sequence of callbacks.
  • **learner_kwargs: Keyword arguments that will be internally passed to Learner.

Returns

A fastai Learner.