search

wenzhu23333 / Federated-Learning

An open source FL implement with dataset(Femnist, Shakespeare, MNIST, Cifar-10 and Fashion-Mnist) using pytorch
GNU General Public License v3.0
115 stars 27 forks source link

number of client #5

Closed DongYan1121 closed 1 year ago

DongYan1121 commented 1 year ago

Hello, I would like to ask if the number of training users can be added to this

wenzhu23333 commented 1 year ago

For MNIST, Cifar-10, Fashion-MNIST, you can change in the utils/options.py. For FEMNIST, Shakespeare, changing the number of training users is useless, because these datasets are sampled from project leaf, you can sample by yourself. Or you can rewrite code in utils/dataset.py by yourself to fulfill this requirement.