c377363e505d3223df0136f14e415ecd5e2bceaa,rl_coach/graph_managers/graph_manager.py,GraphManager,restore_checkpoint,#GraphManager#,544

Before Change

                screen.warning("No checkpoint to restore in: {}".format(self.task_parameters.checkpoint_restore_dir))
            else:
                screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
                if not hasattr(self.agent_params.memory, "memory_backend_params") or self.agent_params.memory.memory_backend_params.run_type != str(RunType.ROLLOUT_WORKER):
                    self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)

            [manager.restore_checkpoint(self.task_parameters.checkpoint_restore_dir) for manager in self.level_managers]

    def _get_checkpoint_state_tf(self):
        import tensorflow as tf

After Change

                screen.warning("No checkpoint to restore in: {}".format(self.task_parameters.checkpoint_restore_dir))
            else:
                screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
                self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)

            [manager.restore_checkpoint(self.task_parameters.checkpoint_restore_dir) for manager in self.level_managers]

    def _get_checkpoint_state_tf(self):

Instances