python运行报错***

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

***解决办法：
1、将所设的batch_size改小一点。因为是从github上copy的代码，所以跑不通其实挺正常，毕竟电脑配置啥的，都与作者的不一样。因此我们改成和自己电脑相匹配的就行。

将batch_size改成32，就成功运行。
大家遇到这种情况根据自身情况，改小一点即可。

2、根据自己的数据集改成相应的num_classes.
eg.你的数据集为CIFAR100，有100类，则num_classes=100。