How did you initialize self.split and what are you returning in the __len__ method?
Could you print the length of labels:
split = ... # set the same split value here
print(len(dataset.labels[split]))
How did you initialize self.split and what are you returning in the __len__ method?
Could you print the length of labels:
split = ... # set the same split value here
print(len(dataset.labels[split]))