Normalize giving an error

[shreygupta2809]

RESIZE = (128, 171)
MEAN = [0.43216, 0.394666, 0.37645]
STD = [0.22803, 0.22145, 0.216989]
CROP_SIZE = 112

transform = Compose(
                [
                    # Resize(RESIZE),
                    # CenterCrop(CROP_SIZE),
                    # ToTensor(),
                    Normalize(MEAN, STD)
                ]
            )

On trying the above code (even without commenting) on google collab I get the following error:

Does anyone know what the issue is or how to fix this? Attaching list of pip packages in colab for reference if needed packages.txt

Thanks