赞
踩
在运行代码前边加上:
if __name__ == '__main__':
就可以了
根据评论区,如果torch.utils.data.DataLoader中的num_workers错误 将num_workers改为0即可!