lightning

[source]

ModelAdapter

icevision.models.ross.efficientdet.lightning.model_adapter.ModelAdapter(model, metrics=None)

Lightning module specialized for EfficientDet, with metrics support.

The methods forward, training_step, validation_step, validation_epoch_end are already overriden.

Arguments

• model torch.nn.modules.module.Module: The pytorch model to use.
• metrics List[icevision.metrics.metric.Metric]: Sequence of metrics to use.

Returns

A LightningModule.

---