Heavy-ball method in nonconvex optimization problems