wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-12 12:05:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

flush logger

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
agent/DDPG_agent.py
+4 -2
View File
@@ -67,8 +67,8 @@ class DDPGAgent:
# tensorboard logging
suffix = 'test_' if deterministic else ''
config.logger.scalar_summary(suffix+'action', action, self.total_steps)
config.logger.scalar_summary(suffix+'reward', reward, self.total_steps)
config.logger.scalar_summary(suffix + 'action', action, self.total_steps)
config.logger.scalar_summary(suffix + 'reward', reward, self.total_steps)
for key in info:
config.logger.scalar_summary('info_' + key, info[key], self.total_steps)
@@ -125,4 +125,6 @@ class DDPGAgent:
self.soft_update(self.target_network, self.worker_network)
config.logger.writer.file_writer.flush()
return total_reward, steps