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