pytorch训练模型常见错误（图像）

1、如下图所示，数据加载时：

这个是transform有问题，我遇到了以下情况：
（1）随机裁剪的尺度大于了原图，
（2）PIL读取的图片再次转为PIL格式。
还有可能是图片路径问题：可以调试查看原因。

2、CUDNN_STATUS_NOT_INITIALIZED

（1）batch_size太大，显存不够。
（2） 检查GPU服务器的CUDA版本，一定要在Pytorch官网安装对应你CUDA版本的Pytorch。
（3） 若提示loss不能反传，则检查分类类别数量是否与实际数据的类别数量一致。