Method __exit__

lib/model_wrapper.py:85–88

(self, exc_type, exc_val, exc_tb)

                module.mask = self.mask

    def __exit__(self, exc_type, exc_val, exc_tb):
        for name, module in self.model.named_modules():
            if isinstance(module, ActLinear):
                module.mask = None


def make_Act(model, verbose=False):