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

Before Change

            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:
            self.logger.info("[{0}] Model saved as <{1}>".format(self.name, model_name))

    def load(self, model_name=None, save_dir="checkpoints", verbose=True):

After Change

            os.makedirs(model_dir)

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

Instances