nlvr2_head
allennlp_models.vision.models.heads.nlvr2_head
Nlvr2Head#
@Head.register("nlvr2")
class Nlvr2Head(Head):
| def __init__(
| self,
| vocab: Vocabulary,
| embedding_dim: int,
| label_namespace: str = "labels"
| )
forward#
class Nlvr2Head(Head):
| ...
| 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,
| label: Optional[torch.Tensor] = None,
| label_weights: Optional[torch.Tensor] = None
| ) -> Dict[str, torch.Tensor]
get_metrics#
class Nlvr2Head(Head):
| ...
| def get_metrics(self, reset: bool = False) -> Dict[str, float]
make_output_human_readable#
class Nlvr2Head(Head):
| ...
| def make_output_human_readable(
| self,
| output_dict: Dict[str, torch.Tensor]
| ) -> Dict[str, torch.Tensor]
default_predictor#
class Nlvr2Head(Head):
| ...
| default_predictor = "nlvr2"