The forward expects a NestedTensor, which consists of: - samples.tensor: batched images, of shape [batch_size x 3 x H x W] - samples.mask: a binary mask of shape [batch_size x H x W], containing 1 on padded pixels It returns a dict with the following elements:
(self, samples: NestedTensor, captions, encode_and_save=True,
memory_cache=None, paf_samples=None, arm_query=None,
img_names=None, encodings_of_tokenized=None)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected