deepmd.tf.entrypoints.freeze
Script for freezing TF trained graph so it can be used with LAMMPS and i-PI.
Module Contents
- deepmd.tf.entrypoints.freeze.freeze(*, checkpoint_folder: str, output: str, node_names: str | None = None, nvnmd_weight: str | None = None, **kwargs) → None[source]
Freeze the graph in supplied folder.
- Parameters:
- checkpoint_folder
str location of either the folder with checkpoint or the checkpoint prefix
- output
str output file name
- node_names
Optional[str], optional names of nodes to output, by default None
- nvnmd_weight
Optional[str], optional nvnmd weight file
- **kwargs
other arguments
