federated-learning
federated-learning copied to clipboard
A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561
python main_fed.py --dataset mnist --iid --num_channels 1 --model cnn --epochs 50 --gpu 0 In addition Hi, about main_fed.py, how to run the program results for non-iid data
how to acquire the middle gradient of each client in FL by using pytorch? I try hook, but can't figure out
我對參數做出微調為了可以符合colab 所需要的參數
Can I experiment on other tasks? For example, some tasks in NLP.
Why does the FedAvg use a simple average without weight?
Hi, When I ran your code locally, I found that the program reported an error when downloading the test dataset. This dataset website can't be accessed normally.  
Please tell me What is "the parameters C=0.1, B=10, E=5"? and the "C=0.1, B=10, E=5" which parameters represent in the options.py, Thanks.
Hello. Thanks for you nice code. But I think the accuracy can be better with the new 'tranform' of cifar: ``` trans_train = transforms.Compose([ transforms.RandomCrop(32, padding=4), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize((0.4914, 0.4822,...
Dear, First thank you for your code. I have run your code, however, the result is not satisfying. Result: Training accuracy: 43.00 Testing accuracy: 43.00 ## my cmd: > python...
I think it's wrong when the data distribution is noniid, should change to: def FedAvg(w, dict_len): w_avg = copy.deepcopy(w[0]) for k in w_avg.keys(): w_avg[k] = w_avg[k] * dict_len[0] for i...