Skip to content

visual_entailment_head

allennlp_models.vision.models.heads.visual_entailment_head

[SOURCE]


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]