grad¶
allgrad¶
- allgrad(model)¶
return true if all parameters in the model require grad.
- Return type
bool
anygrad¶
- anygrad(model)¶
return true if any parameters in the model require grad.
- Return type
bool
freeze¶
- freeze(model)¶
freeze a model in place
- Parameters
model (
Module) – model to freeze- Return type
Module
named_trainable_parameters¶
- named_trainable_parameters(model)¶
yields named trainable params from a model
- Parameters
model (
Module) – model to get named params from- Yields
tuples of name (str), param (nn.Parameter)
- Return type
Iterator[Tuple[str,Parameter]]
requires_grad¶
- requires_grad(param)¶
return true if the parameter requires grad
- Return type
bool
trainable_parameters¶
- trainable_parameters(model)¶
yields trainable parameters from a model.
- Return type
Iterator[Parameter]
unfreeze¶
- unfreeze(model)¶
unfreeze a model in place
- Parameters
model (
Module) – model to unfreeze- Return type
Module