code: def get_imagenet(
      datapath, split, batch_size, shuffle, transform=TRANSFORM
  ):
      ds = torchvision.datasets.ImageNet(
          root=datapath, split=split, transform=transform
      )
      loader = torch.utils.data.DataLoader(
          ds, shuffle=shuffle, batch_size=batch_size
      )
      return ds, loader
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_imagenet(datapath, split, batch_size, shuffle, transform) and returns: ds,  loader