dhuseljic
Results
2
comments of
dhuseljic
Same problem here.
You could use the subset class `torch.utils.data.Subset(dataset, indices=train_idx)`.