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

model aggregate? #4

Closed b856741 closed 2 years ago

b856741 commented 2 years ago

你好，請問這邊所用的模型聚合的方法為何? 因為似乎沒有看到有關模型聚合的相關參數

wenzhu23333 commented 2 years ago

在models的Fed.py里面，只是简单的将模型平均了一下，可以在这里修改，如果改成按照数据集加权平均也可以。