Skip to content

fastai

[source]

learner

icevision.models.ross.efficientdet.fastai.learner.learner(dls, model, cbs=None, **learner_kwargs)

Fastai Learner adapted for EfficientDet.

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.