fixed convergence and finishing of training