Skip to content

Commit

Permalink
add missing spaces in doc
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz authored Mar 2, 2024
1 parent 0959888 commit d29a310
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions deepmd/utils/argcheck.py
Original file line number Diff line number Diff line change
Expand Up @@ -458,8 +458,8 @@ def descrpt_se_atten_args():
doc_tebd_dim = "The dimension of atom type embedding."
doc_temperature = "The scaling factor of normalization in calculations of attention weights, which is used to scale the matmul(Q, K)."
doc_scaling_factor = (

Check warning on line 460 in deepmd/utils/argcheck.py

View check run for this annotation

Codecov / codecov/patch

deepmd/utils/argcheck.py#L458-L460

Added lines #L458 - L460 were not covered by tests
"The scaling factor of normalization in calculations of attention weights, which is used to scale the matmul(Q, K)."
"If `temperature` is None, the scaling of attention weights is (N_hidden_dim * scaling_factor)**0.5."
"The scaling factor of normalization in calculations of attention weights, which is used to scale the matmul(Q, K). "
"If `temperature` is None, the scaling of attention weights is (N_hidden_dim * scaling_factor)**0.5. "
"Else, the scaling of attention weights is setting to `temperature`."
)
doc_normalize = (

Check warning on line 465 in deepmd/utils/argcheck.py

View check run for this annotation

Codecov / codecov/patch

deepmd/utils/argcheck.py#L465

Added line #L465 was not covered by tests
Expand Down

0 comments on commit d29a310

Please sign in to comment.