pytorch數據加載器
阿新 • • 發佈:2019-04-23
一個 last tro ase int pin bsp 如果 sampler
class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, num_workers=0, collate_fn=<function default_collate>, pin_memory=False, drop_last=False)
參數:
- dataset (Dataset) – 加載數據的數據集。
- batch_size (int, optional) – 每個batch加載多少個樣本(默認: 1)。
- shuffle (bool, optional) – 設置為
True
- sampler (Sampler, optional) – 定義從數據集中提取樣本的策略。如果指定,則忽略
shuffle
參數。 - num_workers (int, optional) – 用多少個子進程加載數據。0表示數據將在主進程中加載(默認: 0)
- collate_fn (callable, optional) –
- pin_memory (bool, optional) –
- drop_last (bool, optional) – 如果數據集大小不能被batch size整除,則設置為True後可刪除最後一個不完整的batch。如果設為False並且數據集的大小不能被batch size整除,則最後一個batch將更小。(默認: False)
pytorch數據加載器