你遇到的这个错误：不是你的代码问题，而是 Colab + Python 3.12 + datasets 的一个已知兼容性 bug。

根本原因是：

• datasets 在把样本转换成 PyTorch tensor 时，会尝试导入 torchvision.io.VideoReader

• 但 Colab 默认安装的 torchvision 不包含 VideoReader

• 于是触发 ImportError，哪怕你根本没用视频数据

解决方案非常简单：禁止 datasets 使用 torchvision。