code: def __init__(
      self,
      img_text_pairs1,
      img_text_pairs2,
      similarity_scores,
      process_batch,
  ):
      super().__init__()
      self.texts1 = [t1 for i1, t1 in img_text_pairs1]
      self.texts2 = [t2 for i2, t2 in img_text_pairs2]
  
      self.imgs1 = [i1 for i1, t1 in img_text_pairs1]
      self.imgs2 = [i2 for i2, t2 in img_text_pairs2]
  
      self.similarity_scores = similarity_scores
  
      self.process_batch = process_batch
  
imports: import torch
  from torch.utils.data import Dataset
  from PIL import Image
  import numpy
label_desc: Use image node_id 1 to create a new ImageTextPairDataset object from inherited base classes: Dataset with example: obj = ImageTextPairDataset(img_text_pairs1, img_text_pairs2, similarity_scores, process_batch)