From 55191dd32194f715b178a14f53f5595929f6c612 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 12 Jul 2021 23:51:05 -0400 Subject: [PATCH 1/2] also convert input v1 to v2 for dp compress --- deepmd/entrypoints/compress.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/entrypoints/compress.py b/deepmd/entrypoints/compress.py index 1222b1e51b..c18162ab6f 100644 --- a/deepmd/entrypoints/compress.py +++ b/deepmd/entrypoints/compress.py @@ -6,7 +6,7 @@ from deepmd.common import j_loader from deepmd.utils.argcheck import normalize -from deepmd.utils.compat import convert_input_v0_v1 +from deepmd.utils.compat import updata_deepmd_input from .freeze import freeze from .train import train @@ -64,7 +64,7 @@ def compress( """ jdata = j_loader(INPUT) if "model" not in jdata.keys(): - jdata = convert_input_v0_v1(jdata, warning=True, dump="input_v1_compat.json") + jdata = updata_deepmd_input(jdata, warning=True, dump="input_v1_compat.json") jdata["model"]["compress"] = {} jdata["model"]["compress"]["type"] = 'se_e2_a' jdata["model"]["compress"]["compress"] = True From 4630abe56baf5bddea86e3e39c2f4cac2d667758 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 13 Jul 2021 00:06:13 -0400 Subject: [PATCH 2/2] correct the dump file name --- deepmd/entrypoints/compress.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/entrypoints/compress.py b/deepmd/entrypoints/compress.py index bdcdfb57f1..6b85999426 100644 --- a/deepmd/entrypoints/compress.py +++ b/deepmd/entrypoints/compress.py @@ -65,7 +65,7 @@ def compress( """ jdata = j_loader(INPUT) if "model" not in jdata.keys(): - jdata = updata_deepmd_input(jdata, warning=True, dump="input_v1_compat.json") + jdata = updata_deepmd_input(jdata, warning=True, dump="input_v2_compat.json") jdata["model"]["compress"] = {} jdata["model"]["compress"]["type"] = 'se_e2_a' jdata["model"]["compress"]["compress"] = True