unitorch.cli.models

ClassificationOutputs

Bases: TensorOutputs, WriterMixin

Outputs for classification models.

outputs instance-attribute

outputs: Tensor

ClassificationTargets

Bases: TensorTargets

Targets for classification models.

targets instance-attribute

targets: Tensor

sample_weight class-attribute instance-attribute

sample_weight: Optional[Tensor] = tensor(1.0)

EmbeddingOutputs

Bases: TensorOutputs, WriterMixin

Outputs for embedding models, with up to four optional auxiliary embeddings.

embedding instance-attribute

embedding: Tensor

embedding1 class-attribute instance-attribute

embedding1: Optional[Tensor] = empty(0)

embedding2 class-attribute instance-attribute

embedding2: Optional[Tensor] = empty(0)

embedding3 class-attribute instance-attribute

embedding3: Optional[Tensor] = empty(0)

embedding4 class-attribute instance-attribute

embedding4: Optional[Tensor] = empty(0)

GenerationOutputs

Bases: TensorOutputs, WriterMixin

Outputs for generation models.

sequences instance-attribute

sequences: Tensor

sequences_scores class-attribute instance-attribute

sequences_scores: Optional[Tensor] = empty(0)

GenerationTargets

Bases: TensorTargets

Targets for generation models.

refs instance-attribute

refs: Tensor

masks class-attribute instance-attribute

masks: Optional[Tensor] = empty(0)

sample_weight class-attribute instance-attribute

sample_weight: Optional[Tensor] = empty(0)

LossOutputs

Bases: TensorOutputs

loss instance-attribute

loss: Tensor