You can form a list with this format. [(Img1, grasp1),(Img1, grasp2), (Img1, grasp3),…(Imgn, graspn)]. Iterate over this list using dataloader. If you need more information let me know. I can add a base code.