Solved by ptrblck in post #4 
              
              
                You could try something like this: 
class MyDataset(Dataset):
    def __init__(self, numpy_arr, mean, std):
        self.data = torch.from_numpy(numpy_arr)
        self.mean = mean
        self.std = std
        #normalize here or in __getitem__

    def __getitem__(self, index):
        data = self…