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

Before Change

            os.makedirs(model_dir)

        // Create Saver
        with self.graph.as_default():
            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

After Change

        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.")

Instances