b56a4b355dfc302fa449d223e065af3a7acac986,snorkel/learning/pytorch/noise_aware_model.py,TorchNoiseAwareModel,load,#TorchNoiseAwareModel#Any#Any#Any#,252

Before Change

        model_dir = os.path.join(save_dir, model_name)

        // Load model kwargs needed to rebuild model
        with open(os.path.join(model_dir, "model_kwargs.pkl"), "rb") as f:
            model_kwargs = load(f)
        
        // Create new graph, build network, and start session
        self._build_new_graph_session(**model_kwargs)

        // Initialize variables
        with self.graph.as_default():

After Change

        Load model from file and rebuild in new graph / session.
        model_name = model_name or self.name
        model_dir = os.path.join(save_dir, model_name)
        warnings.warn("Unstable! Please extensively test this part of the code when time permits")
        
        self.load_state_dict(
            torch.load("{}/model.params".format(model_dir))
        )

Instances