search

CharlieDinh / pFedMe

Personalized Federated Learning with Moreau Envelopes (pFedMe) using Pytorch (NeurIPS 2020)
290 stars 88 forks source link

modify niid #13

Closed DarlingHang closed 3 years ago

DarlingHang commented 3 years ago

Hello, Charlie! Thanks for the fantastic paper and codes. I am trying to generate some niid data with different properties. When running data/Mnist/generate_niid_20users.py, I came into several errors. There seem to be some issues with the name "idx". I redefined it and modified the index of mnist_data[l]. Also, the implementation of the distribution of labels to users has seemed to be slightly incorrect, leading to an error later.

CharlieDinh commented 3 years ago

Than DarlingHang. I merged it into the repo.