错误原因：torch.utils.data.DataLoader中的num_workers错误
将num_workers改为0即可，0是默认值。
num_workers是用来指定开多进程的数量，默认值为0，表示不启用多进程。
若：
将num_workers设置为0，程序报错，并提示设置环境变量KMP_DUPLICATE_LIB_OK=TRUE
那你可以在设置环境变量KMP_DUPLICATE_LIB_OK=TRUE
或者使用临时环境变量：（在代码开始处添加这行代码）
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
版权声明：本文为weixin_43917589原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。