Solved by mariosasko in post #2 
              
              
                You can read about the CIFAR10 class frequencies on the offical website (they are already well knowned). 
It’s much better to iterate over torch.utils.data.Dataset directly instead of DataLoader (you don’t need for your task batching, parallelization and other utilities that this class provides). 
T…