mml.core.models.smp

class SMPGenericModel[source]

Bases: BaseModel

__init__(**kwargs)[source]

forward(x: Tensor) → Dict[str, Tensor][source]

forward_features(x: Tensor) → Tensor[source]

supports(task_type: TaskType) → bool[source]: SMP support classification and segmentation tasks.

class SMPHead[source]

Bases: BaseHead

__init__(task_type: TaskType, num_classes: int, num_features: int)[source]

forward(x: Tensor) → Tensor[source]