In the BaseDataSets method under dataloader/dataset, there is no 'fold' parameter, but this parameter is passed in during training. Is this a typo?
