为节约内存,深度学习必须使用分流的方式加载数据
keras 使用的是fit_generator
Pytorch 使用DataLoader
__len__()
函数可以控制数据加载的大小;- 使用
collate_fn
得到自定义输出:
注意 :(可以将一个batch的数据整合成想要的输入)
为节约内存,深度学习必须使用分流的方式加载数据
keras 使用的是fit_generator
Pytorch 使用DataLoader
__len__()
函数可以控制数据加载的大小;collate_fn
得到自定义输出: