Closed denadai2 closed 4 years ago
train / valid splitting of BaseDataLoader
in this project, or torch.utils.data.random_split
only split the indices of dataset. I think the best method of doing this is to have different dataset object for train and validation set.
training=True
in __init__
, and use different data by this flag.train_dataset
and valid_dataset
respectively.self.valid_loader = torch.utils.data.DataLoader(valid_dataset, ...)
split_validation
method to just return self.valid_loader
Thank you
Hi, I would like to remove some of the transformations in the data loader when there is the validation phase. Any suggestion to do it in a clean way through your template?
thanks