diff --git a/deepmd/dpmodel/descriptor/dpa2.py b/deepmd/dpmodel/descriptor/dpa2.py index d42a9da26e..86ed7ec40c 100644 --- a/deepmd/dpmodel/descriptor/dpa2.py +++ b/deepmd/dpmodel/descriptor/dpa2.py @@ -476,6 +476,7 @@ def init_subclass_params(sub_data, sub_class): self.repformers.dim_in, bias=False, precision=precision, + seed=child_seed(seed, 3), ) self.tebd_transform = None if self.add_tebd_to_repinit_out: @@ -484,6 +485,7 @@ def init_subclass_params(sub_data, sub_class): self.repformers.dim_in, bias=False, precision=precision, + seed=child_seed(seed, 4), ) assert self.repinit.rcut > self.repformers.rcut assert self.repinit.sel[0] > self.repformers.sel[0]