c12f538d2d496f5e8f932abc1ee72a586b9ea855,fonduer/learning/disc_learning.py,TFNoiseAwareModel,save,#TFNoiseAwareModel#Any#Any#Any#Any#,330

Before Change

            saver = tf.train.Saver(tf.global_variables())

        // Save model kwargs needed to rebuild model
        with open(os.path.join(model_dir, "model_kwargs.pkl"), "wb") as f:
            dump(self.model_kwargs, f)

        // Save graph and report if verbose
        saver.save(
            self.session, os.path.join(model_dir, model_name), global_step=global_step
        )
        if verbose:

After Change

        if not os.path.exists(model_dir):
            os.makedirs(model_dir)

        params = {
            "model": self.state_dict(),
            "cardinality": self.cardinality,
            "name": model_name,
            "config": self.model_kwargs,
            "epoch": global_step,
        }

        model_file = "{0}.mdl.ckpt.{1}".format(model_name, global_step)

        try:
            torch.save(params, "{0}/{1}".format(model_dir, model_file))
        except BaseException:
            self.logger.warning("Saving failed... continuing anyway.")

        if verbose:

Instances