b56a4b355dfc302fa449d223e065af3a7acac986,snorkel/learning/pytorch/noise_aware_model.py,TorchNoiseAwareModel,save,#TorchNoiseAwareModel#Any#Any#Any#Any#,225

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

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

After Change

        warnings.warn("Unstable! Please extensively test this part of the code when time permits")
        // implement Path here
        print(model_dir)
        torch.save(
            self.state_dict(), 
            "{}/model.params".format(model_dir)
        )

        if verbose:
            print("[{0}] Model saved as <{1}>".format(self.name, model_name))
        

Instances