MCPcopy Create free account
hub / github.com/huggingface/transformers / freeze_params

Function freeze_params

examples/seq2seq/utils.py:239–241  ·  view source on GitHub ↗
(model: nn.Module)

Source from the content-addressed store, hash-verified

237
238
239def freeze_params(model: nn.Module):
240 for par in model.parameters():
241 par.requires_grad = False
242
243
244def grad_status(model: nn.Module) -> Iterable:

Callers 6

__init__Method · 0.90
sanity_check_gradientsMethod · 0.90
freeze_embedsMethod · 0.90
sanity_check_gradientsMethod · 0.90
__init__Method · 0.90
freeze_embedsMethod · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected