Add init-frz-model command support to the frozen model generated by t…

…his command (deepmodeling#1137)
njzjz · Sep 14, 2021 · dae4068 · dae4068
1 parent df10d84
commit dae4068
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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)
+        return tf.import_graph_def(graph_def, input_map = feed_dict, return_elements = return_elements, name = "")
diff --git 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__(