gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 19:50:17 +01:00
Code Issues Projects Releases Wiki Activity

update CartPole_PPO

Browse Source
This commit is contained in:
zach dwiel
2018-10-11 12:37:09 -04:00
parent 3e5e5475de
commit def76b4cc6
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rl_coach/presets/CartPole_PPO.py
View File

@@ -58,9 +58,9 @@ agent_params.pre_network_filter.add_observation_filter('observation', 'normalize
env_params = Mujoco()
env_params.level = 'CartPole-v0'
vis_params = VisualizationParameters()
vis_params.video_dump_methods = [SelectedPhaseOnlyDumpMethod(RunPhase.TEST), MaxDumpMethod()]
vis_params.dump_mp4 = False
visualization_params = VisualizationParameters()
visualization_params.video_dump_methods = [SelectedPhaseOnlyDumpMethod(RunPhase.TEST), MaxDumpMethod()]
visualization_params.dump_mp4 = False
########
# Test #
@@ -71,5 +71,5 @@ preset_validation_params.min_reward_threshold = 150
preset_validation_params.max_episodes_to_achieve_reward = 250
graph_manager = BasicRLGraphManager(agent_params=agent_params, env_params=env_params,
schedule_params=schedule_params, vis_params=vis_params,
schedule_params=schedule_params, visualization_params=visualization_params,
preset_validation_params=preset_validation_params)