Skip to content

nlvr2_head

allennlp_models.vision.models.heads.nlvr2_head

[SOURCE]


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"