Hi smth,
I use this script to train the model, https://github.com/lynic/notebooks/blob/master/ml/pytorch/CIFAR10_CNN.ipynb 156
EPOCHES I set to 20, and I use pytorch 0.1.12 with cuda 8.0 .
Then I use the code in question description to test the model, seems batch_size really impact the results.