解决CUDA error: an illegal memory

神奇的是，减少batchsize报错就消失了，甚至其他数据集都不会报错。

查看num_classes是否与当前数据集类别数一致！改正，解决！