摘要: 自定义 数据集对象 在 pytorch 中,数据加载可以通过自定义数据集对象实现; 数据集对象被抽象为 DataSet 类; 自定义数据集对象,需要继承该类,并且实现 __getitem__ 和 __len__ 两个方法 示例 class DogCat(data.Dataset): def __in 阅读全文