federated-model-aggregation
federated-model-aggregation copied to clipboard
Allow developers to pass extra parameters to aggregation methods
Is your feature request related to a problem? Please describe. The current method of aggregation on allows for users to average the gradient updates from all clients and apply a global update.
Describe the outcome you'd like: I would like functionality to pass in parameters to the aggregation methods. For example, client weights, to weight the averaging of gradient updates.
Additional context: