Cambios para entrenar la red X veces

Agregue los cambios para que podamos hacer que la red se entrene X
cantidad de veces asi es más facil hacer las pruebas.