torch.utils.data.DataLoader
數據加載器,結合了數據集和取樣器
在訓練模型時使用到此函數,用來把訓練數據分成多個小組,此函數每次拋出一組數據。直至把所有的數據都拋出。就是做一個數據的初始化。
torch.utils.data.DataLoader( dataset,#數據加載 batch_size = 1,#批處理大小設置 shuffle = False,#是否進項洗牌操作 sampler = None,#指定數據加載中使用的索引/鍵的序列 batch_sampler = None,#和sampler類似 num_workers = 0,#是否進行多進程加載數據設置 collate_fn = None,#是否合並樣本列表以形成一小批Tensor pin_memory = False,#如果True,數據加載器會在返回之前將Tensors復制到CUDA固定內存 drop_last = False,#True如果數據集大小不能被批處理大小整除,則設置為刪除最后一個不完整的批處理。 timeout = 0,#如果為正,則為從工作人員收集批處理的超時值 worker_init_fn = None )
dataloader = torch.utils.data.DataLoader(imgs, batch_size=batch_size)