Preserve tensor learning rates across scheduler updates (#8202)