Method __init__

lib/model_wrapper.py:76–78

(self, model, mask)

class set_mask:
    def __init__(self, model, mask):
        self.model = model
        self.mask = mask

    def __enter__(self):
        for name, module in self.model.named_modules():