Preserve tensor learning rates in OneCycle (#8205)