From c2f306cb6f7f01013e70d648063ac4e4025403ba Mon Sep 17 00:00:00 2001 From: Duo <50307526+iProzd@users.noreply.github.com> Date: Tue, 27 Feb 2024 15:32:40 +0800 Subject: [PATCH] Update __init__.py --- deepmd/pt/model/model/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py index 38ee01ce44..cbd44d9bbb 100644 --- a/deepmd/pt/model/model/__init__.py +++ b/deepmd/pt/model/model/__init__.py @@ -135,7 +135,7 @@ def get_spin_model(model_params): model_params["fitting_net"]["exclude_types"] = atom_exclude_types # descriptor model_params["descriptor"]["ntypes"] = ntypes - descriptor = Descriptor(**model_params["descriptor"]) + descriptor = BaseDescriptor(**model_params["descriptor"]) # fitting fitting_net = model_params.get("fitting_net", None) fitting_net["type"] = fitting_net.get("type", "ener")