Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
Han Wang committed Mar 15, 2024
1 parent 8d05a4e commit 1cb163a
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 deletions deepmd/dpmodel/atomic_model/base_atomic_model.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -105,9 +105,9 @@ def forward_common_atomic(
ret_dict
dict of output atomic properties.
should implement the definition of `fitting_output_def`.
ret_dit["mask"] of shape nf x nloc will be provided.
ret_dit["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real.
ret_dit["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual.
ret_dict["mask"] of shape nf x nloc will be provided.
ret_dict["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real.
ret_dict["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual.
"""
_, nloc, _ = nlist.shape
Expand Down
6 changes: 3 additions & 3 deletions deepmd/pt/model/atomic_model/base_atomic_model.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -132,9 +132,9 @@ def forward_common_atomic(
ret_dict
dict of output atomic properties.
should implement the definition of `fitting_output_def`.
ret_dit["mask"] of shape nf x nloc will be provided.
ret_dit["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real.
ret_dit["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual.
ret_dict["mask"] of shape nf x nloc will be provided.
ret_dict["mask"][ff,ii] == 1 indicating the ii-th atom of the ff-th frame is real.
ret_dict["mask"][ff,ii] == 0 indicating the ii-th atom of the ff-th frame is virtual.
"""
_, nloc, _ = nlist.shape
Expand Down

0 comments on commit 1cb163a

Please sign in to comment.