Optimal learning rate schedules in high-dimensional non-convex optimization problems