federated-learning icon indicating copy to clipboard operation
federated-learning copied to clipboard

A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561

Results 17 federated-learning issues
Sort by recently updated
recently updated
newest added

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. ![QQ图片20210319135120](https://user-images.githubusercontent.com/24271582/111737443-992ba100-88ba-11eb-91ad-6673e0eea2fd.png) ![QQ图片20210319135133](https://user-images.githubusercontent.com/24271582/111737490-ae083480-88ba-11eb-9082-352e4a27426a.png)

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...