记一次CUDA报错

报错内容：CUDA error: device-side assert triggered

原因：使用ResNet50训练时使用了pretrained=True的模型，但实际类别数classes远超过MNIST自带的1000上限，导致标签越界。

解决方法：改成使用pretrained=False的模型，并指定num_classes