在加载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 就可以解决了

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐