In the 1st case, I have images of many different classes A,B,C,D,E in the test data.
In the Dataloader, as the shuffle is set to True, there are images of different classes in the batch of 10.
Here, the accuracy is about 90%.
In the 2nd case, I have the images of only class A in the test data.
In the Dataloader, even if the shuffle is set to True, all the images are of the same class in the batch of 10.
But in the second case, it comes out wrong very often. The accuracy is only about 10%.