From c5b2028a23160af931e48a0a4c5dec7be779f30e Mon Sep 17 00:00:00 2001 From: denghuilu Date: Mon, 13 Sep 2021 22:47:55 +0800 Subject: [PATCH] Add init-frz-model command support to the frozen model generated by this command --- deepmd/model/ener.py | 2 +- deepmd/model/tensor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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__(