deepmd.tf.entrypoints.freeze

Script for freezing TF trained graph so it can be used with LAMMPS and i-PI.

References

https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc

Functions

freeze(*, checkpoint_folder, output[, node_names, ...])

Freeze the graph in supplied folder.

Module Contents

deepmd.tf.entrypoints.freeze.freeze(*, checkpoint_folder: str, output: str, node_names: str | None = None, nvnmd_weight: str | None = None, **kwargs)

Freeze the graph in supplied folder.

Parameters:

checkpoint_folderstr

location of either the folder with checkpoint or the checkpoint prefix

outputstr

output file name

node_namesOptional[str], optional

names of nodes to output, by default None

nvnmd_weightOptional[str], optional

nvnmd weight file

**kwargs

other arguments