pytorch Dataloader 数据集加载

为节约内存,深度学习必须使用分流的方式加载数据
keras 使用的是fit_generator
Pytorch 使用DataLoader

  1. __len__()函数可以控制数据加载的大小;
  2. 使用collate_fn得到自定义输出:
    注意 :(可以将一个batch的数据整合成想要的输入)

你可能感兴趣的:(Pytorch)