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

deal with non scalar action

Browse Source
This commit is contained in:
Mike C
2018-01-22 10:48:40 +08:00
parent 6a92f52b4f
commit 74788bcf96
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agent/DDPG_agent.py
+4 -1
View File
@@ -81,7 +81,10 @@ class DDPGAgent:
# tensorboard logging
suffix = 'test_' if deterministic else ''
config.logger.scalar_summary(suffix + 'action', action, self.total_steps)
if action.squeeze().ndim == 0:
config.logger.scalar_summary(suffix + 'action', action, self.total_steps)
else:
config.logger.histo_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)