Solved by ptrblck in post #2 
              
              
                I’m not sure there is a clean way of handling different subsets within a single Dataset class. 
If you want to handle the split yourself, I would rather create a custom function or class (if statefull) and return the corresponding dataset using e.g. TensorDataset or Subset. 
E.g. this might be a usa…