RuntimeError: Given groups=1, weight of size 40 1 5 5, expected input[4, 3, 28, 28]

在将MNIST数据集替换为其他数据集时，出现了如下错误：
RuntimeError: Given groups=1, weight of size 40 1 5 5, expected input[4, 3, 28, 28] to have 1 channels, but got 3 channels instead
原因是，MNIST数据集是灰度图片，而我输入的是彩色图片，有RGB三个通道。
解决:使用 transforms.Grayscale()函数

def loadtraindata():
    path = r"/home/ouc/zhangshan/train"                                          # 路径
    trainset = torchvision.datasets.ImageFolder(path,
                                                transform=transforms.Compose([
                                                transforms.Resize((28,28)),  # 将图片缩放到指定大小（h,w）或者保持长宽比并缩放最短的边到int大小
                                                transforms.CenterCrop(28),
                                                transforms.Grayscale(),#将图片变成灰度图片
                                                transforms.ToTensor()])
                                                )

    trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,shuffle=True, num_workers=4)
    return trainloader