torch.utils.data.DataLoader 將數據按批次分成很多組,每次拋出一個小組


torch.utils.data.DataLoader

數據加載器,結合了數據集和取樣器
在訓練模型時使用到此函數,用來把訓練數據分成多個小組,此函數每次拋出一組數據。直至把所有的數據都拋出。就是做一個數據的初始化。

torch.utils.data.DataLoader(
                            dataset,#數據加載
                            batch_size = 1,#批處理大小設置
                            shuffle = False,#是否進項洗牌操作
                            sampler = None,#指定數據加載中使用的索引/鍵的序列
                            batch_sampler = None,#和sampler類似
                            num_workers = 0,#是否進行多進程加載數據設置
                            collat​​e_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)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM