摘要: 上文介绍了数据读取、数据转换、批量处理等等。了解到在PyTorch中,数据加载主要有两种方式: 1. 自定义的数据集对象。数据集对象被抽象为Dataset类,实现自定义的数据集需要继承Dataset。且须实现__len__()和__getitem__()两个方法。 2. 利用torchvision包 阅读全文
posted @ 2018-03-30 10:32 三年一梦 阅读(4838) 评论(0) 推荐(1) 编辑