We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bbb2a49 commit aa5ced2Copy full SHA for aa5ced2
1 file changed
deepmd/pt/model/atomic_model/dp_atomic_model.py
@@ -64,7 +64,8 @@ def __init__(
64
self.rcut = self.descriptor.get_rcut()
65
self.sel = self.descriptor.get_sel()
66
self.fitting_net = fitting
67
- self.fitting_net.reinit_exclude(self.atom_exclude_types)
+ if hasattr(self.fitting_net, "reinit_exclude"):
68
+ self.fitting_net.reinit_exclude(self.atom_exclude_types)
69
super().init_out_stat()
70
self.enable_eval_descriptor_hook = False
71
self.enable_eval_fitting_last_layer_hook = False
0 commit comments