From 2bd94737b51e038afc145abb51efebcb9be80aa5 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Thu, 9 May 2024 22:58:30 +0800
Subject: [PATCH] delete other unused params

---
 deepmd/tf/env.py     | 26 --------------------------
 doc/freeze/freeze.md |  4 ----
 2 files changed, 30 deletions(-)

diff --git a/deepmd/tf/env.py b/deepmd/tf/env.py
index 4c20a2b978..cdb4feadc3 100644
--- 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
index 4d0c5c1d2b..3bc8cd6452 100644
--- 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.
-
-:::
-
-::::