pytorch
涤生843
这个作者很懒,什么都没留下…
展开
-
DataLoader类详解
该类实现了__iter__函数,所以它是一个可迭代对象。每次迭代返回的就是从dataset中取得batch_size大小的数据data(注意,这个data的长度不是batch_sieze,而是dataset的__getitem__函数返回值的数目。例如,如果__getitem__函数返回的是(img,target),则len(data)是2)而dataset如果实现了__getitem__函数,那么就可以通过查看它的__getitem__函数返回值,来得知最后data中包含什么内容。原创 2023-02-04 16:33:47 · 2086 阅读 · 0 评论 -
DataLoader模块实现迭代的过程简析
DataLoader类虽然只实现了__iter__方法。但是它在__iter__方法中,通过__get_iterator()返回一个_SingleProcessDataLoader对象,该对象是一个迭代器,所以DataLoader类的对象是一个可迭代对象。原创 2023-02-03 18:19:50 · 404 阅读 · 0 评论 -
关于pytorch的一些操作——如何查看官方API
2.进入相应的包后,再去搜索该类,否则会出现找不到的情况。原创 2023-02-03 16:03:34 · 392 阅读 · 0 评论