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.

Create new instance of ClsOut(logits, probs)

logits: Tensor

Alias for field number 0

probs: Tensor

Alias for field number 1