pytorch之DataLoader()函数
摘要:在训练神经网络时,最好是对一个batch的数据进行操作,同时还需要对数据进行shuffle和并行加速等。对此,PyTorch提供了DataLoader帮助我们实现这些功能。 DataLoader的函数定义如下: dataset:加载的数据集(Dataset对象) batch_size:batch s
阅读全文
posted @ 2019-08-04 15:28
浙公网安备 33010602011771号