ValueError: <PIL.Image.Image image mode=RGB size=28x28 at 0x192F390D430> is not a valid Transform

想做一个自己训练图片分类的程序,代码如图

img

img

img

img

img

img

img


代码参考了这个链接https://blog.csdn.net/qq_53345829/article/details/124308515,本身的图片集是已经按类别分好的0,1,2....,9的图片
报错:
ValueError: <PIL.Image.Image image mode=RGB size=28x28 at 0x192F390D430> is not a valid Transform
该如何修改呢

Transform