Restore correct update_parameters method in MultiTaskTopLayer