deepmd.jax.entrypoints.freeze

Freeze utilities for the JAX backend.

Functions

freeze(→ None)

Freeze a JAX checkpoint into a serialized model file.

Module Contents

deepmd.jax.entrypoints.freeze.freeze(*, checkpoint_folder: str, output: str, **kwargs: object) → None

Freeze a JAX checkpoint into a serialized model file.

Parameters:

checkpoint_folderstr

Location of either the checkpoint directory or a folder containing the stable checkpoint pointer.

outputstr

Output model filename or prefix. The JAX model suffix is added when the filename has no supported backend suffix.

**kwargs

Other CLI arguments accepted for backend entry-point compatibility.