diff --git a/deepmd/model/ener.py b/deepmd/model/ener.py index 441709caff..c179f00f7b 100644 --- a/deepmd/model/ener.py +++ b/deepmd/model/ener.py @@ -270,4 +270,4 @@ def build (self, def _import_graph_def_from_frz_model(self, frz_model, feed_dict, return_elements): graph, graph_def = load_graph_def(frz_model) - return tf.import_graph_def(graph_def, input_map = feed_dict, return_elements = return_elements) \ No newline at end of file + return tf.import_graph_def(graph_def, input_map = feed_dict, return_elements = return_elements, name = "") \ No newline at end of file diff --git a/deepmd/model/tensor.py b/deepmd/model/tensor.py index 2a63eda4d6..17b9ace29d 100644 --- a/deepmd/model/tensor.py +++ b/deepmd/model/tensor.py @@ -191,7 +191,7 @@ def build (self, def _import_graph_def_from_frz_model(self, frz_model, feed_dict, return_elements): graph, graph_def = load_graph_def(frz_model) - return tf.import_graph_def(graph_def, input_map = feed_dict, return_elements = return_elements) + return tf.import_graph_def(graph_def, input_map = feed_dict, return_elements = return_elements, name = "") class WFCModel(TensorModel): def __init__(