pytorch自定义数据读取
pytorch的数据读取主要跟三个类有关: 1. Dataset 2. DataLoader 3. DataLoaderIter定义自己的Dataset类定义自己的Dataset类(继承torch.utils.data.Dataset)并实现两个成员方法 1. __getitem__() 2. __len__()class MyDataset(data.Dataset...