Fix lr schedule
Call lr scheduler before training so that the update takes place properly also if a device missed out one or several training rounds. This happends e.g. when changing the server device in Swarm SL or neglecting client updates if using a loss threshold. Also added lr scheduler for client for parallel SL.