Hi @ptrblck, Thanks so much for the sample code. I followed the steps that you mentioned. But I am still not getting well distributed classes: The following is the class count. This dataset is well distributed, but still I am not getting good distribution:
Class count:
tensor([1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000,  945])
batch index 0, 0/1: 9/6/6,9
batch index 1, 0/1: 6/12/12,4
batch index 2, 0/1: 9/8/9,12
batch index 3, 0/1: 10/6/12,5
batch index 4, 0/1: 5/5/9,12