Pytorch入門-dataloader

DataLoader(dataset, batch_size=1, shuffle=False, sampler=None,
           batch_sampler=None, num_workers=0, collate_fn=None,
           pin_memory=False, drop_last=False, timeout=0,
           worker_init_fn=None, *, prefetch_factor=2,
           persistent_workers=False)

1. dataset (Dataset): 提供資料的資料集物件。
2. batch_size (int, optional): 每個批次中包含的樣本數。預設為 1。
3. shuffle (bool, optional): 是否在每個 epoch 開始時打亂資料。預設為 False。
4. sampler (Sampler, optional): 定義從資料集中抽取樣本的策略。如果指定，shuffle 必須為 False。
5. batch_sampler (Sampler, optional): 與 sampler 類似，但直接返回批次的索引，而不是單個樣本的索引。如果指定，batch_size, shuffle, sampler, 和 drop_last 必須不被指定。
6. num_workers (int, optional): 用於資料載入的子程序數。0 表示資料將在主程序中載入。預設為 0。
7. drop_last (bool, optional): 如果資料集大小不能被 batch_size 整除，設定為 True 後將丟棄最後一個不完整的批次。預設為 False。