UFDModelOutput

class UFDModelOutput(loss: Optional[torch.Tensor] = None, logits: Optional[torch.Tensor] = None)[source]

Base class for outputs of UFD models

Parameters

  • loss (torch.Tensor of shape (1,), optional, returned when labels is provided) – Classification loss. Loss function used is dependent on what is specified in UFDConfig

  • logits (torch.Tensor of shape (batch_size, 1)) – Classification scores.