visual_entailment_head
allennlp_models.vision.models.heads.visual_entailment_head
VisualEntailmentHead#
@Head.register("visual_entailment")
class VisualEntailmentHead(Head):
| def __init__(
| self,
| vocab: Vocabulary,
| embedding_dim: int,
| label_namespace: str = "labels"
| )
forward#
class VisualEntailmentHead(Head):
| ...
| @overrides
| def forward(
| self,
| encoded_boxes: torch.Tensor,
| encoded_boxes_mask: torch.Tensor,
| encoded_boxes_pooled: torch.Tensor,
| encoded_text: torch.Tensor,
| encoded_text_mask: torch.Tensor,
| encoded_text_pooled: torch.Tensor,
| pooled_boxes_and_text: torch.Tensor,
| labels: Optional[torch.Tensor] = None,
| label_weights: Optional[torch.Tensor] = None
| ) -> Dict[str, torch.Tensor]
get_metrics#
class VisualEntailmentHead(Head):
| ...
| @overrides
| def get_metrics(self, reset: bool = False) -> Dict[str, float]