About pytorch

1、DataLoader

DataLoader加载一个Dataset类的对象,自定义的Dataset需要实现__getitem__和__len__两个方法,分别返回数据集中的一个数据和数据集的长度。

原文地址:https://www.cnblogs.com/lxc1910/p/12447317.html