vis4d.model.cls.common

Common types for classification models.

Classes

ClsOut(logits, probs)

Output of the classification results.

class ClsOut(logits: Tensor, probs: Tensor)[source]

Output of the classification results.

logits: Tensor: Alias for field number 0

probs: Tensor: Alias for field number 1