Solved by tom in post #2 
              
              
                So random_split returns to Subset instances that themselves refer to the (single!) DataSet instance. 
The modern workflow for this is to override the collate function in the dataloader (calling default collate and then optionally augmenting the dataset). 
Or you could make full instances of your da…