Convergence Enhancement for Neural Network Integrated Power System Time Domain Simulation