Then instead of for (input, labels) in loaders[split]: you might want to do something like for (input_m, labels_m), (input_c, labels_c) in zip(loaders[split], loaders_cifar[split]): and therefore also instead of pred = net(input) something like pred = net(input_m, input_c).
Regards,
Unity05