About pytorch

1、DataLoader

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

posted @ 2020-03-09 10:59  bloglxc  阅读(92)  评论(0编辑  收藏  举报