I want to train each batch to have the same class. For example, when batch size is 32, update once with 32 data of class 0, and then learn with 32 data of randomly selected class.
If you learn all data in class 0 and then learn about class 1, it can’t be learning.