def __init__(self, model):
    self.model = model
    self.emb_backup = {}
    self.grad_backup = {}
def __init__(self, model):
    self.model = model
    self.emb_backup = {}
    self.grad_backup = {}