These are the loaders of MNIST
# Define loaders
from torch.utils.data import DataLoader
train_loader = DataLoader(train_set, batch_size=64, num_workers=2, shuffle=True, drop_last=True)
val_loader   = DataLoader(val_set,   batch_size=64, num_workers=2, shuffle=False, drop_last=False)
test_loader  = DataLoader(test_set,  batch_size=64, num_workers=2, shuffle=False, drop_last=False)

# Define dictionary of loaders
loaders = {"train": train_loader,
           "val": val_loader,
           "test": test_loader}

And these of CIFAR
# Define loaders
from torch.utils.data import DataLoader
train_loader_cifar = DataLoader(train_set_cifar, batch_size=64, num_workers=2, shuffle=True, drop_last=True)
val_loader_cifar   = DataLoader(val_set_cifar,   batch_size=64, num_workers=2, shuffle=False, drop_last=False)
test_loader_cifar  = DataLoader(test_set_cifar,  batch_size=64, num_workers=2, shuffle=False, drop_last=False)

# Define dictionary of loaders
loaders_cifar = {"train": train_loader_cifar,
           "val": val_loader_cifar,
           "test": test_loader_cifar}