使用pycharm时报错：Cannot find reference ‘data‘ in ‘init.py‘

最新推荐文章于 2024-05-09 20:15:54 发布

原创最新推荐文章于 2024-05-09 20:15:54 发布 · 1.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #人工智能

机器学习专栏收录该内容

7 篇文章

订阅专栏

在PyCharm中遇到数据引用错误时，可以导入torch.utils.data模块并直接使用DataLoader。通过引入DataLoader，为训练集和测试集创建数据加载器，设置batch_size为64，启用shuffle并配置num_workers为2，以优化数据读取。

部署运行你感兴趣的模型镜像

# # # 使用pycharm时报错：Cannot find reference 'data' in 'init.py'

解决办法：

将data导入模块

import torch.utils.data

直接import DataLoader

from torch.utils.data import DataLoader

train_loader = torch.utils.data.DataLoader(train_data, batch_size=64,
                                          shuffle=True, num_workers=2)
test_loader = torch.utils.data.DataLoader(test_data, batch_size=64,
                                          shuffle=True, num_workers=2)

您可能感兴趣的与本文相关的镜像