We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 24325c1 commit 8c52a7dCopy full SHA for 8c52a7d
1 file changed
deepmd/pt/model/atomic_model/dp_atomic_model.py
@@ -154,7 +154,8 @@ def change_type_map(
154
)
155
self.fitting_net.change_type_map(type_map=type_map)
156
# Reinitialize fitting to get correct sel_type
157
- self.fitting_net.reinit_exclude(self.atom_exclude_types)
+ if hasattr(self.fitting_net, "reinit_exclude"):
158
+ self.fitting_net.reinit_exclude(self.atom_exclude_types)
159
160
def has_message_passing(self) -> bool:
161
"""Returns whether the atomic model has message passing."""
0 commit comments