在运行代码前边加上：

if __name__ == '__main__':

就可以了

 

根据评论区，如果torch.utils.data.DataLoader中的num_workers错误 将num_workers改为0即可！

# python
Logo
https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

Claude Code + Chrome：浏览器层面的AI编程新范式已至

Claude Code + Chrome：AI编程新纪元 核心亮点 深度集成：Claude Code实现对Chrome浏览器的完整控制，从URL操作到DOM解析 高效调试：原生支持控制台错误捕获、操作录屏和性能分析 无缝认证：自动调用Chrome会话信息，无需重复登录验证 环境搭建 安装Claude Code CLI（≥2.0.72） 添加Claude in Chrome扩展（≥1.0.36） 执

avatar AI编程社区
cover

C++AI大模型接入SDK—环境搭建

avatar AI编程社区
cover

开源项目“AI思维圆桌”的介绍和对于当前AI编程的思考

avatar AI编程社区