delete other unused params

deepmodeling · May 9, 2024 · 2bd9473 · 2bd9473
1 parent c70666a
commit 2bd9473
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 30 deletions.
diff --git a/deepmd/tf/env.py b/deepmd/tf/env.py
@@ -111,7 +111,6 @@ def dlopen_library(module: str, filename: str):
     "EMBEDDING_NET_PATTERN",
     "TYPE_EMBEDDING_PATTERN",
     "ATTENTION_LAYER_PATTERN",
-    "REMOVE_SUFFIX_DICT",
     "TF_VERSION",
     "tf_py_version",
 ]
@@ -209,31 +208,6 @@ def dlopen_library(module: str, filename: str):
     )
 )
 
-REMOVE_SUFFIX_DICT = {
-    "model_attr/sel_type_{}": "model_attr/sel_type",
-    "model_attr/output_dim_{}": "model_attr/output_dim",
-    "_{}/": "/",
-    # when atom_ener is set
-    "_{}_1/": "_1/",
-    "o_energy_{}": "o_energy",
-    "o_force_{}": "o_force",
-    "o_virial_{}": "o_virial",
-    "o_atom_energy_{}": "o_atom_energy",
-    "o_atom_virial_{}": "o_atom_virial",
-    "o_dipole_{}": "o_dipole",
-    "o_global_dipole_{}": "o_global_dipole",
-    "o_polar_{}": "o_polar",
-    "o_global_polar_{}": "o_global_polar",
-    "o_rmat_{}": "o_rmat",
-    "o_rmat_deriv_{}": "o_rmat_deriv",
-    "o_nlist_{}": "o_nlist",
-    "o_rij_{}": "o_rij",
-    "o_dm_force_{}": "o_dm_force",
-    "o_dm_virial_{}": "o_dm_virial",
-    "o_dm_av_{}": "o_dm_av",
-    "o_wfc_{}": "o_wfc",
-}
-
 
 def set_mkl():
     """Tuning MKL for the best performance.

diff --git a/doc/freeze/freeze.md b/doc/freeze/freeze.md
@@ -30,7 +30,3 @@ $ dp --pt freeze -o model_branch1.pth --head CHOSEN_BRANCH
 ```
 
 The output model is called `model_branch1.pth` being the specific frozen model with `CHOSEN_BRANCH` head.
-
-:::
-
-::::
-Original file line number
+Diff line change
@@ Expand Up / @@ -30,7 +30,3 @@ $ dp --pt freeze -o model_branch1.pth --head CHOSEN_BRANCH @@
     ```
     The output model is called `model_branch1.pth` being the specific frozen model with `CHOSEN_BRANCH` head.
-    :::
-    ::::