一文弄懂Pytorch的DataLoader, DataSet, Sampler之间的关系
以下内容都是针对Pytorch 1.0-1.1介绍。很多文章都是从Dataset等对象自下往上进行介绍,但是对于初学者而言,其实这并不好理解,因为有的时候会不自觉地陷入到一些细枝末节中去,而不能把握重点,所以本文将会自上而下地对Pytorch数据读取方法进行介绍。没错就是下面一行。我们已经拿到了indices,那么下一步我们只需要根据index对数据进行读取即可了。再下面的综上可以知道DataLoader,Sampler和Dataset三者关系如下:在阅读后文的过程中,你始终需要将上面的关系记在心里,这样能帮助你更好地理解。要更加细致地理解Sampler原理,我们需要先阅读一下DataLoader 的源代码,如下:可以看到初始化参数里有两种sampler:Pytorch中已经实现的
下载地址
用户评论