2032abb5ed9c7bc96703f631359858f3b6ae51c6,snorkel/learning/disc_learning.py,TFNoiseAwareModel,load,#TFNoiseAwareModel#Any#Any#,91

Before Change

            @verbose: be talkative?
        
        self.load_info(model_name)
        load_dict = self._build()
        saver = tf.train.Saver(load_dict)
        ckpt = tf.train.get_checkpoint_state("./")
        if ckpt and ckpt.model_checkpoint_path:
            saver.restore(self.session, ckpt.model_checkpoint_path)

After Change

            @verbose: be talkative?
        
        self.load_info(model_name)
        self._build()
        load_dict = self.save_dict or tf.global_variables()
        saver = tf.train.Saver(load_dict)
        ckpt = tf.train.get_checkpoint_state("./")
        if ckpt and ckpt.model_checkpoint_path:

Instances