core.utils.torchutils.set_requires_grad_flag

core.utils.torchutils.set_requires_grad_flag(
    model,
    name_startswith=None,
    name_contains=None,
    requires_grad=True,
)

Set param.requires_grad of all model parameters with a name starting with name_startswith, or name containing name_contains, to requires_grad.