6506f3a125e001e7667771f3c1f570c53dbefd06,tgen/seq2seq.py,Seq2SeqGen,save_to_file,#Seq2SeqGen#Any#,290

Before Change

            self._training_pass(iter_no)

    def save_to_file(self, fname):
        raise NotImplementedError()

    def generate_tree(self, inputs):
        raise NotImplementedError()

After Change

    def save_to_file(self, model_fname):
        log_info("Saving generator to %s..." % model_fname)
        with file_stream(model_fname, "wb", encoding=None) as fh:
            data = {"emb_size": self.emb_size,
                    "batch_size": self.batch_size,
                    "randomize": self.randomize,
                    "passes": self.passes,
                    "da_embs": self.da_embs,
                    "tree_embs": self.tree_embs,
                    "da_dict_size": self.da_dict_size,
                    "tree_dict_size": self.tree_dict_size,
                    "max_da_len": self.max_da_len,
                    "max_tree_len": self.max_tree_len}
            pickle.dump(data, fh, protocol=pickle.HIGHEST_PROTOCOL)
        tf_session_fname = re.sub(r"(.pickle)?(.gz)?$", ".tfsess", model_fname)
        self.saver.save(self.session, tf_session_fname)

    @staticmethod

Instances