5332013bd17636a0d21d189ec21e8254a1d3f4e7,rl_coach/graph_managers/graph_manager.py,GraphManager,save_checkpoint,#GraphManager#,581

Before Change

        checkpoint_path = os.path.join(self.task_parameters.checkpoint_save_dir,
                                       filename)
        if not os.path.exists(os.path.dirname(checkpoint_path)):
            os.mkdir(os.path.dirname(checkpoint_path))  // Create directory structure
        if not isinstance(self.task_parameters, DistributedTaskParameters):
            saved_checkpoint_path = self.checkpoint_saver.save(self.sess, checkpoint_path)
        else:

After Change

        if not os.path.exists(self.task_parameters.checkpoint_save_dir):
            os.mkdir(self.task_parameters.checkpoint_save_dir)  // Create directory structure

        if self.checkpoint_state_updater is None:
            self.checkpoint_state_updater = CheckpointStateUpdater(self.task_parameters.checkpoint_save_dir)

        checkpoint_name = "{}_Step-{}.ckpt".format(
            self.checkpoint_id, self.total_steps_counters[RunPhase.TRAIN][EnvironmentSteps])
        checkpoint_path = os.path.join(self.task_parameters.checkpoint_save_dir, checkpoint_name)

Instances