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

Update for latest gym

Browse Source
This commit is contained in:
Shangtong Zhang
2018-02-09 22:02:58 -07:00
parent e7f036bc6b
commit d4aa8d4a22
6 changed files with 36 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/misc.py
-7
View File
@@ -7,7 +7,6 @@
import numpy as np
import pickle
import os
import gym.monitoring
def run_episodes(agent):
config = agent.config
@@ -31,12 +30,6 @@ def run_episodes(agent):
agent_type, config.tag, agent.task.name), 'wb') as f:
pickle.dump([steps, rewards], f)
if config.render_episode_freq and ep % config.render_episode_freq == 0:
video_recoder = gym.monitoring.VideoRecorder(
env=agent.task.env, base_path='./data/video/%s-%s-%s-%d' % (agent_type, config.tag, agent.task.name, ep))
agent.episode(True, video_recoder)
video_recoder.close()
if config.episode_limit and ep > config.episode_limit:
break