PyTorch--报错“RuntimeError: running_mean should contain *** elements not ***”

保错信息：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()
])