Alternatively you could create a  Dataset  instance for each split (and add the noise to the training dataset), create the split indices, and wrap the datasets together with the corresponding split indices in  Subset s.

This is what I am looking for but I don’t know how to do this…