class
CrossViTForImageClassification
extends
PretrainedModelClassificationHeadMixinCrossViTForImageClassification(config: CrossViTConfig)CrossViT with two per-branch linear heads, output averaged.
Paper §3.3 — the final prediction is the mean of the two branch-specific logits, an implicit two-model ensemble.
Used by 2
Constructors
1Instance methods
1forward(x: Tensor, labels: Tensor | None = None)