1.为什么要重写: 满足自己的需求 2.怎么重写 参考教程 3.注意点 1)Dataset是一个包装类,用来将数据包装为Dataset类,然后传入DataLoader中,在训练模型时使用到此函数,用来把训练数据分成多个小组,此函数每次抛出一组数据。直至把所有的数据都抛出。就是做一个数据的初始化。 2)DataLoader函数中有一个参数:默认drop_last = False,可以手动设置为Ture,这样就会舍弃不能整除batch_size的数据