wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-08-07 11:08:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add Normalized Actions

Browse Source
This commit is contained in:
pranz24
2019-05-20 13:29:23 +05:30
parent 7c5f0cc3b2
commit e9cc3fd7e8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+2 -2
View File
@@ -123,10 +123,10 @@ for i_episode in itertools.count(1):
avg_reward /= episodes
writer.add_scalar('reward/test', episode_reward, i_episode)
writer.add_scalar('reward/test', avg_reward, i_episode)
print("----------------------------------------")
print("Test Episodes: {}, Avg. Reward: {}".format(i_episode, round(episode_reward, 2)))
print("Test Episodes: {}, Avg. Reward: {}".format(episodes, round(avg_reward, 2)))
print("----------------------------------------")
env.close()