Convergence of a Normal Map-based Prox-SGD Method under the KL Inequality