保错信息：RuntimeError: running_mean should contain 3 elements not 1
解决：因为模型的输入是灰度图像输入，也就是输入size为（batch_size,1,H,W）,可是我实际输入模型的图片是RGB三通道图像，所以报错，将RGB转成灰度图就好啦。
转化灰度图代码
from torchvision import transforms transform = transforms.Compose([ transforms.Grayscale(num_output_channels=1), # 彩色图像转灰度图像num_output_channels默认1 transforms.ToTensor() ])
PyTorch——报错“RuntimeError: running_mean should contain *** elements not ***”
原文：https://www.cnblogs.com/zukang/p/14743921.html