Sorry, I mean.
torch.utils.data.DataLoader(dataset, batch_size=32, num_workers=8)
under if __name__ == '__main__'



 mobassir94:

train_loader, val_loader = prepare_dataloader(train, trn_idx, val_idx, data_root='c:/cassava-leaf-disease-classification/train_images/')