I\'m training a neural network (doesn\'t matter which one) on CIFAR-10 dataset. I\'m using Federated Learning: