From dcf7cce30d8cb69ccea8dd9aa66dc2cb97a27edd Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Fri, 13 Sep 2019 19:48:51 -0400 Subject: [PATCH] Flush out print text so I can keep track of the rewards by SSHing in --- rltorch/env/simulate.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rltorch/env/simulate.py b/rltorch/env/simulate.py index ee3b16d..a792ccb 100644 --- a/rltorch/env/simulate.py +++ b/rltorch/env/simulate.py @@ -17,7 +17,7 @@ def simulateEnvEps(env, actor, config, total_episodes = 1, memory = None, logger if episode % config['print_stat_n_eps'] == 0: print("episode: {}/{}, score: {}" - .format(episode, total_episodes, episode_reward)) + .format(episode, total_episodes, episode_reward), flush=True) if logger is not None: logger.append(name + '/EpisodeReward', episode_reward) @@ -51,7 +51,7 @@ class EnvironmentRunSync(): if done: if self.episode_num % self.config['print_stat_n_eps'] == 0: print("episode: {}/{}, score: {}" - .format(self.episode_num, self.config['total_training_episodes'], self.episode_reward)) + .format(self.episode_num, self.config['total_training_episodes'], self.episode_reward), flush=True) if self.logwriter is not None: logger.append(self.name + '/EpisodeReward', self.episode_reward) @@ -92,10 +92,10 @@ class EnvironmentEpisodeSync(): if self.episode_num % self.config['print_stat_n_eps'] == 0: print("episode: {}/{}, score: {}" - .format(self.episode_num, self.config['total_training_episodes'], episodeReward)) + .format(self.episode_num, self.config['total_training_episodes'], episodeReward), flush=True) if self.logwriter is not None: logger.append(self.name + '/EpisodeReward', episodeReward) self.logwriter.write(logger) - self.episode_num += 1 \ No newline at end of file + self.episode_num += 1