Skip to content

Commit fe6a92e

Browse files
committed
Update dpa3.py
1 parent 527cb85 commit fe6a92e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/pt/model/descriptor/dpa3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def share_params(self, base_class, shared_level, resume=False) -> None:
269269
# share all parameters in type_embedding, repflow
270270
if shared_level == 0:
271271
self._modules["type_embedding"] = base_class._modules["type_embedding"]
272-
self.repflows.share_params(base_class.repflow, 0, resume=resume)
272+
self.repflows.share_params(base_class.repflows, 0, resume=resume)
273273
# shared_level: 1
274274
# share all parameters in type_embedding
275275
elif shared_level == 1:

0 commit comments

Comments
 (0)