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

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, united_model: bool = False, **kwargs)[source]

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

united_modelbool

when in multi-task mode, freeze all nodes into one unit model

**kwargs

other arguments