Changed the number of iterations for the training module