解决AttributeError: module ‘torch.utils.data‘ has no attribute ‘Dataloader‘问题

在加载pytorch数据集时候 莫名其妙抛出一个错误
AttributeError: module ‘torch.utils.data’ has no attribute ‘Dataloader’

很无语。。。。
经过我多次查找资料 得出结论
很简单的问题。

train_loader = torch.utils.data.Dataloader(dataset=train_dateset,
                                           batch_size = batch_size,
                                           shuffle=True
                                           )

里面的Dataloader改成 DataLoader 就可以解决了

你可能感兴趣的:(pytorch,深度学习,神经网络)