However I need the train_dataset_subset to return something like this:
Counter ({ 0: 300, 1: 300, 2: 300, … , 9: 300})
The reason it is 300, is because it is 3000 split over the 10 labels. How would I go about doing this? I specifically want the subset, not the original dataset.
You won’t be able to directly access it and would need to iterate the Subset since the samples will be created from the passed indices while the samples are drawn.