deepmd.tf.utils.convert

Attributes

log

Functions

detect_model_version(input_model)	Detect DP graph version.
convert_to_21(→ None)	Convert DP graph to 2.1 graph.
convert_13_to_21(→ None)	Convert DP 1.3 graph to 2.1 graph.
convert_12_to_21(→ None)	Convert DP 1.2 graph to 2.1 graph.
convert_10_to_21(→ None)	Convert DP 1.0 graph to 2.1 graph.
convert_012_to_21(→ None)	Convert DP 0.12 graph to 2.1 graph.
convert_20_to_21(→ None)	Convert DP 2.0 graph to 2.1 graph.
convert_pb_to_pbtxt(→ None)	Convert DP graph to graph text.
convert_pbtxt_to_pb(→ None)	Convert DP graph text to graph.
convert_dp012_to_dp10(→ None)	Convert DP 0.12 graph text to 1.0 graph text.
convert_dp10_to_dp11(→ None)	Convert DP 1.0 graph text to 1.1 graph text.
convert_dp12_to_dp13(→ None)	Convert DP 1.2 graph text to 1.3 graph text.
convert_dp13_to_dp20(→ None)	Convert DP 1.3 graph text to 2.0 graph text.
convert_dp20_to_dp21(→ None)

Module Contents

deepmd.tf.utils.convert.log

deepmd.tf.utils.convert.detect_model_version(input_model: str)

Detect DP graph version.

Parameters:

input_modelstr

filename of the input graph

deepmd.tf.utils.convert.convert_to_21(input_model: str, output_model: str, version: str | None = None) → None

Convert DP graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

versionstr

version of the input graph, if not specified, it will be detected automatically

deepmd.tf.utils.convert.convert_13_to_21(input_model: str, output_model: str) → None

Convert DP 1.3 graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

deepmd.tf.utils.convert.convert_12_to_21(input_model: str, output_model: str) → None

Convert DP 1.2 graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

deepmd.tf.utils.convert.convert_10_to_21(input_model: str, output_model: str) → None

Convert DP 1.0 graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

deepmd.tf.utils.convert.convert_012_to_21(input_model: str, output_model: str) → None

Convert DP 0.12 graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

deepmd.tf.utils.convert.convert_20_to_21(input_model: str, output_model: str) → None

Convert DP 2.0 graph to 2.1 graph.

Parameters:

input_modelstr

filename of the input graph

output_modelstr

filename of the output graph

deepmd.tf.utils.convert.convert_pb_to_pbtxt(pbfile: str, pbtxtfile: str) → None

Convert DP graph to graph text.

Parameters:

pbfilestr

filename of the input graph

pbtxtfilestr

filename of the output graph text

deepmd.tf.utils.convert.convert_pbtxt_to_pb(pbtxtfile: str, pbfile: str) → None

Convert DP graph text to graph.

Parameters:

pbtxtfilestr

filename of the input graph text

pbfilestr

filename of the output graph

deepmd.tf.utils.convert.convert_dp012_to_dp10(file: str) → None

Convert DP 0.12 graph text to 1.0 graph text.

Parameters:

filestr

filename of the graph text

deepmd.tf.utils.convert.convert_dp10_to_dp11(file: str) → None

Convert DP 1.0 graph text to 1.1 graph text.

Parameters:

filestr

filename of the graph text

deepmd.tf.utils.convert.convert_dp12_to_dp13(file: str) → None

Convert DP 1.2 graph text to 1.3 graph text.

Parameters:

filestr

filename of the graph text

deepmd.tf.utils.convert.convert_dp13_to_dp20(fname: str) → None

Convert DP 1.3 graph text to 2.0 graph text.

Parameters:

fnamestr

filename of the graph text

deepmd.tf.utils.convert.convert_dp20_to_dp21(fname: str) → None