Solved by arya47 in post #3 
              
              
                You can also create a Dataset class which returns the same image, len of 1000 for 1000 instances and returns the same image in getitem 
from torch.utils.data import Dataset
class data(Dataset):
    def __init__(self, image_path, transform=None):
        self.image_path = image_path
        self.tran…