Function freeze_params

seq2seq/utils.py:544–547

Set requires_grad=False for each of model.parameters()

(model: nn.Module)

def freeze_params(model: nn.Module):
    """Set requires_grad=False for each of model.parameters()"""
    for par in model.parameters():
        par.requires_grad = False


def freeze_embeds(model):