Generalization Bound of Gradient Descent for Non-Convex Metric Learning