Match parameters for train() with levenberg_marquardt()
Adjust parameters for better convergence