使用torchvision进行CIFAR-10数据增强时出现TypeError错误求助

大家好，我最近在用torchvision给CIFAR-10数据集做数据增强的时候碰到了TypeError，试了好几种调整都没解决，想请社区的大佬们帮忙看看问题出在哪。

先说明下我的环境版本：

• torchvision: 0.15.2+cu117
• torch: 2.0.1+cu117

下面是我写的数据增强代码：

import torchvision.transforms as transforms

strength = 0.2
color_jitter = transforms.ColorJitter(
    brightness = 0.8 * strength, contrast = 0.8 * strength,
    saturation = 0.8 * strength, hue = 0.2 * strength
)

rand_color_jitter = transforms.RandomApply([color_jitter], p = 0.8)

rand_gray = transforms.RandomGrayscale(p = 0.2)

color_distortion = transforms.Compose(
    [
        rand_color_jitter,
        rand_gray
    ]
)

mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225]

train_data_augmentation = transforms.Compose(
    [
        transforms.RandomResizedCrop(size = 32, scale = (0.14, 1.0)),
        transforms.GaussianBlur(kernel_size = (3, 3), sigma = (0.1, 2.0)),
        color_distortion,
        # v2.ToImage(),
        # v2.ToDtype(torch.float32, scale = True),
        # v2.ToDtype(torch.float32),
        # v2.Normalize(),
        transforms.ToTensor(),
        transforms.Normalize(mean=mean, std=std)
    ]
)

我原本尝试过用torchvision v2的一些接口（比如注释掉的那几行），但换成旧版接口后还是触发了TypeError。有没有可能是我的transforms组合顺序或者参数设置有问题？麻烦各位帮忙分析一下，谢谢！

备注：内容来源于stack exchange，提问作者Arun