About pytorch

1、DataLoader

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

Related Posts

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注