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

Revert "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 reverts commit b8d21c73bf.
This commit is contained in:
Gourav Roy
2019-01-02 22:36:05 -08:00
parent 6dd7ae2343
commit 779d3694b4
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,
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__':