Exit training loop if max reward is reached in an episode

[mesjou]

With simple environment max reward is reached after 2 million training steps and training could be finished