Skip to content

PSL performance

Tim Tobias Bauerle requested to merge psl-performance into main

Reduced the size of the gradients sent to the clients after training the server model. Previously, these were sent as views of all gradients, so that each client received data which was the number of clients times larger than necessary. This lead to a significant drop in performance. Now each client receives only its fraction of gradients. Also fixed the case if no gradients are computed for a model parameter.

Edited by Tim Tobias Bauerle

Merge request reports