deepmd.tf2.entrypoints#
Entry points for the TensorFlow 2 backend.
Submodules#
Functions#
| Compress a TF2 training checkpoint and export a SavedModel. |
| Freeze a TF2 training checkpoint into a TensorFlow SavedModel. |
| Run DeePMD model training with TensorFlow 2 eager execution. |
Package Contents#
- deepmd.tf2.entrypoints.enable_compression(input_file: str, output: str, stride: float = 0.01, extrapolate: int = 5, check_frequency: int = -1, training_script: str | None = None, head: str | None = None, **kwargs: Any) None[source]#
Compress a TF2 training checkpoint and export a SavedModel.
- deepmd.tf2.entrypoints.freeze(*, checkpoint_folder: str, output: str, head: str | None = None, **kwargs: Any) None[source]#
Freeze a TF2 training checkpoint into a TensorFlow SavedModel.
- deepmd.tf2.entrypoints.train(*, INPUT: str, init_model: str | None, restart: str | None, output: str, init_frz_model: str | None, mpi_log: str, log_level: int, log_path: str | None, skip_neighbor_stat: bool = False, finetune: str | None = None, model_branch: str = '', use_pretrain_script: bool = False, **kwargs: Any) None[source]#
Run DeePMD model training with TensorFlow 2 eager execution.