mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
Trace tests update
This commit is contained in:
@@ -46,18 +46,12 @@ vis_params = VisualizationParameters()
|
||||
vis_params.video_dump_methods = [SelectedPhaseOnlyDumpMethod(RunPhase.TEST), MaxDumpMethod()]
|
||||
vis_params.dump_mp4 = False
|
||||
|
||||
|
||||
########
|
||||
# Test #
|
||||
########
|
||||
# this preset is too slow to test on a regular basis
|
||||
|
||||
# preset_validation_params = PresetValidationParameters()
|
||||
# preset_validation_params.test = True
|
||||
# preset_validation_params.min_reward_threshold = 150
|
||||
# preset_validation_params.max_episodes_to_achieve_reward = 250
|
||||
|
||||
preset_validation_params = PresetValidationParameters()
|
||||
preset_validation_params.trace_test_levels = ['cartpole:swingup', 'hopper:hop']
|
||||
|
||||
graph_manager = BasicRLGraphManager(agent_params=agent_params, env_params=env_params,
|
||||
schedule_params=schedule_params, vis_params=vis_params,)
|
||||
# preset_validation_params=preset_validation_params)
|
||||
schedule_params=schedule_params, vis_params=vis_params,
|
||||
preset_validation_params=preset_validation_params)
|
||||
|
||||
Reference in New Issue
Block a user