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

Revert "Updated comments"

Browse Source 
This reverts commit 740f7937cd.
This commit is contained in:
Gourav Roy
2019-01-02 22:33:37 -08:00
parent 740f7937cd
commit 2461892c9e
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 # 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] [manager.restore_checkpoint(self.task_parameters.checkpoint_restore_dir) for manager in self.level_managers]
# Recreate session. This will help use the new Tensorflow Graph created in above step. # Recreate the session to use the new TF Graphs
self.create_session(self.task_parameters) self.create_session(self.task_parameters)
if checkpoint is None: if checkpoint is None:
@@ -568,7 +568,7 @@ class GraphManager(object):
screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path)) screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path) self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)
else: else:
# Create new session # Create the session to use the new TF Graphs
self.create_session(self.task_parameters) self.create_session(self.task_parameters)
def _get_checkpoint_state_tf(self): def _get_checkpoint_state_tf(self):