7a9ab625ca95a5951a7cac572d8e4aa36a3afbba,deeppavlov/models/ner/ner_network.py,NerNetwork,load,#NerNetwork#,189

Before Change

    def load(self):
        saver = tf.train.Saver()
        saver.restore(self._sess, str(self.ser_path))

    def tokens_batch_to_numpy_batch(self, batch_x, batch_y=None):
        // Determine dimensions
        batch_size = len(batch_x)

After Change

        Load session from checkpoint
        
        saver = tf.train.Saver()
        ckpt = self.get_checkpoint_state()
        if ckpt and ckpt.model_checkpoint_path:
            print("\n:: restoring checkpoint from", ckpt.model_checkpoint_path, "\n")
            saver.restore(self._sess, ckpt.model_checkpoint_path)
            print("session restored")
        else:
            print("\n:: <ERR> checkpoint not found! \n")

    def tokens_batch_to_numpy_batch(self, batch_x, batch_y=None):
        // Determine dimensions
        batch_size = len(batch_x)
        max_utt_len = max([len(utt) for utt in batch_x])

Instances