pytorch 1.7

加入下边的import 语句 pycharm就能识别data了

import torch.utils.data #如果不用这个就会出现pycharm不识别data的问题

例子:

import torch
import torch.utils.data #如果不用这个就会出现pycharm不识别data的问题
...
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

解决方法来源于https://github.com/pytorch/pytorch/issues/41794

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐