Thank you ! I decided to create the training and validation indices before creating the Dataset. Much simpler this way.
Dataset