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

comment out the part of test in 'test_basic_rl_graph_manager_with_cartpole_dqn_and_repeated_checkpoint_restore' that run in infinite loop

Browse Source
This commit is contained in:
gouravr
2018-12-16 10:56:00 -08:00
parent 1f0980c448
commit b8d21c73bf
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
rl_coach/tests/graph_managers/test_basic_rl_graph_manager.py
View File

@@ -52,17 +52,17 @@ def test_basic_rl_graph_manager_with_cartpole_dqn_and_repeated_checkpoint_restor
graph_manager.create_graph(task_parameters=TaskParameters(framework_type=Frameworks.tensorflow, graph_manager.create_graph(task_parameters=TaskParameters(framework_type=Frameworks.tensorflow,
experiment_path="./experiments/test", experiment_path="./experiments/test",
apply_stop_condition=True)) apply_stop_condition=True))
graph_manager.improve() # graph_manager.improve()
graph_manager.save_checkpoint() # graph_manager.save_checkpoint()
#
graph_manager.task_parameters.checkpoint_restore_dir = "./experiments/test/checkpoint" # graph_manager.task_parameters.checkpoint_restore_dir = "./experiments/test/checkpoint"
graph_manager.agent_params.memory.register_var('memory_backend_params', # graph_manager.agent_params.memory.register_var('memory_backend_params',
MemoryBackendParameters(store_type=None, # MemoryBackendParameters(store_type=None,
orchestrator_type=None, # orchestrator_type=None,
run_type=str(RunType.ROLLOUT_WORKER))) # run_type=str(RunType.ROLLOUT_WORKER)))
while True: # while True:
graph_manager.restore_checkpoint() # graph_manager.restore_checkpoint()
gc.collect() # gc.collect()
if __name__ == '__main__': if __name__ == '__main__':