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

Updated comments

Browse Source
This commit is contained in:
Gourav Roy
2018-12-25 21:52:07 -08:00
parent 73c4c850a5
commit 740f7937cd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/graph_managers/graph_manager.py
View File

@@ -559,7 +559,7 @@ class GraphManager(object):
# As part of this restore, Agent recreates the global, target and online networks
[manager.restore_checkpoint(self.task_parameters.checkpoint_restore_dir) for manager in self.level_managers]
# Recreate the session to use the new TF Graphs
# Recreate session. This will help use the new Tensorflow Graph created in above step.
self.create_session(self.task_parameters)
if checkpoint is None:
@@ -568,7 +568,7 @@ class GraphManager(object):
screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)
else:
# Create the session to use the new TF Graphs
# Create new session
self.create_session(self.task_parameters)
def _get_checkpoint_state_tf(self):