deepmd.tf2.entrypoints.freeze

Freeze utilities for the TensorFlow 2 backend.

Functions

freeze(→ None)	Freeze a TF2 training checkpoint into a TensorFlow SavedModel.
select_model_branch(→ dict[str, Any])	Select one branch from a single-task or multi-task serialized payload.

Module Contents

deepmd.tf2.entrypoints.freeze.freeze(*, checkpoint_folder: str, output: str, head: str | None = None, **kwargs: Any) → None

Freeze a TF2 training checkpoint into a TensorFlow SavedModel.

deepmd.tf2.entrypoints.freeze.select_model_branch(data: dict[str, Any], head: str | None = None) → dict[str, Any]

Select one branch from a single-task or multi-task serialized payload.