efficientnet_b0¶
- lucid.models.efficientnet_b0(num_classes: int = 1000, **kwargs) EfficientNet¶
The efficientnet_b0 function creates an instance of the EfficientNet-B0 model, a lightweight and efficient convolutional neural network preset built from the default EfficientNetConfig values for EfficientNet-B0.
Total Parameters: 5,289,636
Function Signature¶
@register_model
def efficientnet_b0(num_classes: int = 1000, **kwargs) -> EfficientNet:
Parameters¶
- num_classes (int, optional):
The number of output classes for the classification task. Defaults to 1000 (e.g., for ImageNet).
- kwargs (dict, optional):
Additional keyword arguments forwarded to EfficientNetConfig, excluding the preset width_coef, depth_coef, scale, and dropout fields.
Returns¶
- EfficientNet:
An instance of the EfficientNet class constructed from the EfficientNet-B0 preset.