flush logger

This commit is contained in:
Mike C
2018-01-22 10:17:05 +08:00
parent b73f35c727
commit 106ff12873
3 changed files with 8 additions and 2 deletions
+2
View File
@@ -135,4 +135,6 @@ class DeterministicPolicyGradient:
self.shared_reward_normalizer.offline_stats.merge(self.reward_normalizer.online_stats)
self.reward_normalizer.online_stats.zero()
config.logger.writer.file_writer.flush()
return steps, total_reward
+2
View File
@@ -167,4 +167,6 @@ class ProximalPolicyOptimization:
self.critic_opt.step()
config.total_epochs.value += 1
config.logger.writer.file_writer.flush()
return batched_steps, batched_rewards