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