from torch.utils.data import Dataset, DataLoader
时间: 2023-12-11 19:03:36 浏览: 82
pytorch数据读取Dataloader与Dataset
这是 PyTorch 中用于创建自定义数据集和数据加载器的模块。Dataset 用于加载和存储数据集,DataLoader 用于从 Dataset 中获取批量数据。通过自定义 Dataset 和 DataLoader,我们可以灵活地处理不同的数据集,并且能够利用 PyTorch 中的多进程并行加载数据,提高数据加载速度。
阅读全文