摘要:
自定义 数据集对象 在 pytorch 中,数据加载可以通过自定义数据集对象实现; 数据集对象被抽象为 DataSet 类; 自定义数据集对象,需要继承该类,并且实现 __getitem__ 和 __len__ 两个方法 示例 class DogCat(data.Dataset): def __in 阅读全文
posted @ 2020-01-21 16:17
努力的孔子
阅读(377)
评论(0)
推荐(0)
浙公网安备 33010602011771号