Script for freezing TF trained graph so it can be used with LAMMPS and i-PI.Module Contents
Functions
freeze (*, checkpoint_folder, output[, node_names, ...])
| Freeze the graph in supplied folder. |
- deepmd.tf.entrypoints.freeze.freeze(*, checkpoint_folder: str, output: str, node_names: str | None = None, nvnmd_weight: str | None = None, **kwargs)[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