Number of epochs is set to 500, 700, 400