万物小白

摘要： 1. 首先导入需要用到的包 from torch.utils.data import DataLoader,Dataset 2. 自定义Dataset 一般情况下我们使用Dataset，需要自定义一个类来继承Dataset，然后实现__getitem__()方法和__len__()方法 使用示例如下 阅读全文