Fix PSL
Implemented PSL as in the literature, i.e. added an additional gradient averaging step where all clients send the gradients to the server and receive the average gradients.
Implemented PSL as in the literature, i.e. added an additional gradient averaging step where all clients send the gradients to the server and receive the average gradients.