6b33ca24d5df63dcae69f4e6140327f5fecbc897,deeppavlov/core/models/tf_model.py,TFModel,get_checkpoint_state,#TFModel#,87

Before Change

    def get_checkpoint_state(self):
        if self.ser_path.is_dir():
            return tf.train.get_checkpoint_state(self.ser_path)
        else:
            return tf.train.get_checkpoint_state(self.ser_path.parent)

    // @check_path_exists()

After Change

        saver.save(self.sess, save_path)

    def get_checkpoint_state(self):
        if self.load_path:
            if self.load_path.parent.is_dir():
                return tf.train.get_checkpoint_state(self.load_path.parent)
            else:
                warn("Provided `load_path` is incorrect!")
        else:
            warn("No `load_path` is provided for {}".format(self.__class__.__name__)")

    @overrides
    def load(self):
        
        Load session from checkpoint

Instances