Skip to content

outputs

ModelForImageClassificationOutput

Bases: TypedDict

The output for image classification models.

Source code in glasses/models/vision/image/classification/outputs.py 
6
7
8
9
class ModelForImageClassificationOutput(TypedDict):
    """The output for image classification models."""

    logits: Tensor