1. 程式人生 > >pytorch數據加載器

pytorch數據加載器

一個 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
    時會在每個epoch重新打亂數據(默認: False).
  • 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數據加載器