18fba98733551c294c760e7186f6a18d44d73d18,python/baseline/tf/lm/model.py,WordLanguageModel,save_md,#WordLanguageModel#Any#,117

Before Change

        state = {"hsz": self.hsz, "batchsz": self.batchsz, "layers": self.layers}
        with open(basename + ".state", "w") as f:
            json.dump(state, f)

        tf.train.write_graph(self.sess.graph_def, outdir, base + ".graph", as_text=False)
        with open(basename + ".saver", "w") as f:
            f.write(str(self.saver.as_saver_def()))

After Change

            "layers": self.layers,
            "tgt_key": self.tgt_key
        }
        for prop in ls_props(self):
            state[prop] = getattr(self, prop)

        write_json(state, basename + ".state")
        for key, embedding in self.embeddings.items():
            embedding.save_md("{}-{}-md.json".format(basename, key))

Instances