bug fix in TrainUntilConvergence function parameters