torch.utils.data.dataset和torch.utils.data.dataloader 如何使用
from torch.utils.data.dataset import Dataset
from torch.utils.data.dataloader import DataLoader
1⃣️. torch.utils.data.Dataset 可以来构造pytorch数据集。以这个类构造的子类,一定要定义两个函数。
__ len __ | 用于提供数据集size |
---|---|
__ getitem __ | 通过给定索引获取数据和标签。一次只能获取一个数据 |
所以通过torch.utils.data.DataLoader来定义一个新的迭代器,实现batch读取。
2⃣️. torch.utils.data.dataloader