# Optimal PyTorch DataLoader config from torch.utils.data import DataLoader, TensorDataset dataloader = DataLoader( dataset, batch_size=64, shuffle=True, num_workers=4, pin_memory=True, persistent_workers=True, prefetch_factor=2 )