tfm.vision.factory.build_classification_model

Builds the classification model.

tfm.vision.factory.build_classification_model(
    input_specs: tf.keras.layers.InputSpec,
    model_config: tfm.vision.configs.image_classification.ImageClassificationModel,
    l2_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
    skip_logits_layer: bool = False,
    backbone: Optional[tf.keras.Model] = None
) -> tf.keras.Model