李宏毅2022HW3一段代码报错，求指正。

我将李宏毅ML2022Spring的hw3下载下来后，导入Jupyter Notebook进行运行，数据集准备工作都没什么问题。但是，运行到下面的代码时系统报错了。

但是，给的文件上这里显示正常。

这部分代码的上面有下列的代码

这个地方出现了CrossEntropyLoss()，我查了查，这个地方有可能出错，但是不知道该如何改。
求指教！

李宏毅给出的代码样例https://github.com/Fafa-DL/Lhy_Machine_Learning/blob/main/2022%20ML/03%20Images%20input/2022ML_HW3_Image_Classification.ipynb