Function freeze_params

penzai/core/variables.py:854–861

r"""Version of `freeze_variables` that only freezes `Parameter`\ s.

(
    tree: Any,
    predicate: Callable[[Parameter], bool] | None = None,
)

def freeze_params(
    tree: Any,
    predicate: Callable[[Parameter], bool] | None = None,
) -> Any:
  r"""Version of `freeze_variables` that only freezes `Parameter`\ s."""
  return freeze_variables(  # type: ignore
      tree, predicate=_type_filtered_predicate(predicate, Parameter)
  )


@typing.overload