From b8d21c73bf939ee160538082709b281a790618ef Mon Sep 17 00:00:00 2001 From: gouravr Date: Sun, 16 Dec 2018 10:56:00 -0800 Subject: [PATCH] comment out the part of test in 'test_basic_rl_graph_manager_with_cartpole_dqn_and_repeated_checkpoint_restore' that run in infinite loop --- .../test_basic_rl_graph_manager.py | 22 +++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/rl_coach/tests/graph_managers/test_basic_rl_graph_manager.py b/rl_coach/tests/graph_managers/test_basic_rl_graph_manager.py index a572fd9..fb5ba9d 100644 --- a/rl_coach/tests/graph_managers/test_basic_rl_graph_manager.py +++ b/rl_coach/tests/graph_managers/test_basic_rl_graph_manager.py @@ -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, experiment_path="./experiments/test", apply_stop_condition=True)) - graph_manager.improve() - graph_manager.save_checkpoint() - - graph_manager.task_parameters.checkpoint_restore_dir = "./experiments/test/checkpoint" - graph_manager.agent_params.memory.register_var('memory_backend_params', - MemoryBackendParameters(store_type=None, - orchestrator_type=None, - run_type=str(RunType.ROLLOUT_WORKER))) - while True: - graph_manager.restore_checkpoint() - gc.collect() + # graph_manager.improve() + # graph_manager.save_checkpoint() + # + # graph_manager.task_parameters.checkpoint_restore_dir = "./experiments/test/checkpoint" + # graph_manager.agent_params.memory.register_var('memory_backend_params', + # MemoryBackendParameters(store_type=None, + # orchestrator_type=None, + # run_type=str(RunType.ROLLOUT_WORKER))) + # while True: + # graph_manager.restore_checkpoint() + # gc.collect() if __name__ == '__main__':