Thanks I’m still having trouble trying to figure it out. I’m loading my dataset through Imagefolder and iteratin through DataLoader. I added the filename in getitem of Imagefolder but iterating still gives me only target and label