FedML-AI / FedNLP

FedNLP: An Industry and Research Integrated Platform for Federated Learning in Natural Language Processing, Backed by FedML, Inc. The Previous Research Version is Accepted to NAACL 2022
223 stars 45 forks source link

About large model #22

Open xiang-xiang-zhu opened 2 years ago

xiang-xiang-zhu commented 2 years ago

I want to know how do you maintain the parameters of each large model (such as Bert) in the process of federated learning, such as the fedavg algorithm? Because before server aggregation, if you run federated learning locally, you need to save many model parameters in memory

sauravpr commented 2 years ago

From my understanding, they are held in CPU memory of server, and aggregations are carried out using the model state dictionaries