Skip to content

Commit aa5ced2

Browse files
committed
fix bug: reinit_exclude if fitting has this method
1 parent bbb2a49 commit aa5ced2

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

deepmd/pt/model/atomic_model/dp_atomic_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,8 @@ def __init__(
6464
self.rcut = self.descriptor.get_rcut()
6565
self.sel = self.descriptor.get_sel()
6666
self.fitting_net = fitting
67-
self.fitting_net.reinit_exclude(self.atom_exclude_types)
67+
if hasattr(self.fitting_net, "reinit_exclude"):
68+
self.fitting_net.reinit_exclude(self.atom_exclude_types)
6869
super().init_out_stat()
6970
self.enable_eval_descriptor_hook = False
7071
self.enable_eval_fitting_last_layer_hook = False

0 commit comments

Comments
 (0)