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

Upgrade to PyTorch v0.2.0

Browse Source
This commit is contained in:
Shangtong Zhang
2017-10-28 11:06:20 -06:00
parent a59f2d2911
commit 9615195f5e
9 changed files with 27 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/run.py
+1 -1
View File
@@ -38,7 +38,7 @@ def run_episodes(agent):
agent.save('data/%s-%s-model-%s.bin' % (agent_type, config.tag, agent.task.name))
test_rewards = []
for _ in range(config.test_repetitions):
test_rewards.append(agent.episode(True))
test_rewards.append(agent.episode(True)[0])
avg_reward = np.mean(test_rewards)
avg_test_rewards.append(avg_reward)
config.logger.info('Avg reward %f(%f)' % (