Skip to content

Commit

Permalink
remove duplicated convert_13_to_21 (#1986)
Browse files Browse the repository at this point in the history
Strange, I added duplicated `convert_13_to_21` in #1597.

Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz authored Oct 13, 2022
1 parent 77c7ad4 commit 2486918
Showing 1 changed file with 0 additions and 19 deletions.
19 changes: 0 additions & 19 deletions deepmd/utils/convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,25 +4,6 @@
from google.protobuf import text_format


def convert_13_to_21(input_model: str, output_model: str):
"""Convert DP 1.3 graph to 2.1 graph.
Parameters
----------
input_model : str
filename of the input graph
output_model : str
filename of the output graph
"""
convert_pb_to_pbtxt(input_model, 'frozen_model.pbtxt')
convert_dp13_to_dp20('frozen_model.pbtxt')
convert_dp20_to_dp21('frozen_model.pbtxt')
convert_pbtxt_to_pb('frozen_model.pbtxt', output_model)
if os.path.isfile('frozen_model.pbtxt'):
os.remove('frozen_model.pbtxt')
print("the converted output model (2.1 support) is saved in %s" % output_model)


def convert_13_to_21(input_model: str, output_model: str):
"""Convert DP 1.3 graph to 2.1 graph.
Expand Down

0 comments on commit 2486918

Please sign in to comment.