首页 > 其他分享 >12、dataloader的使用

12、dataloader的使用

时间:2023-02-19 14:33:05浏览次数:40  
标签:12 shuffle dataloader 抓取 每次 使用 数据 size

dataloader数据加载器:将数据加载到神经网络中。定义每次取多少数据,怎么取

1、打开pytorch官网----Doc中找pytorch官方文档----搜索dataloader

出现在torch.utils,.data.DataLoader工具包中,

 

 常用参数设置:

  dataset:数据集的位置

  batch_size:每一次传入多少数据

  shuffle:是否将数据打乱,如果设置为true说明每次传入顺序不一样,如果为false则说明每次顺序一样

  num_workers:多进程,默认为0,有个现象:当他设置为大于0时,windows系统可能会出现错误,那把他设回为0.

  drop_last:数据集的最后余留部分是否舍弃?比如总数据集一共200张,每次batch_size为3,那么200/3=66~2,余下的两张是否丢弃?

 

 

 这个抓取是每次随机抓取4张图片。默认是随机

shuffle设置为False时,2轮的for循环抓取都是一样的

 

标签:12,shuffle,dataloader,抓取,每次,使用,数据,size
From: https://www.cnblogs.com/ar-boke/p/17131579.html

相关文章