Skip to content

Commit

Permalink
se_r has been supported
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz authored Feb 27, 2024
1 parent 6718215 commit 8caa6d6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/utils/argcheck.py
Original file line number Diff line number Diff line change
Expand Up @@ -302,7 +302,7 @@ def descrpt_se_a_tpe_args():
]


@descrpt_args_plugin.register("se_e2_r", alias=["se_r"], doc=doc_only_tf_supported)
@descrpt_args_plugin.register("se_e2_r", alias=["se_r"])
def descrpt_se_r_args():
doc_sel = 'This parameter set the number of selected neighbors for each type of atom. It can be:\n\n\
- `List[int]`. The length of the list should be the same as the number of atom types in the system. `sel[i]` gives the selected number of type-i neighbors. `sel[i]` is recommended to be larger than the maximally possible number of type-i neighbors in the cut-off radius. It is noted that the total sel value must be less than 4096 in a GPU environment.\n\n\
Expand Down

0 comments on commit 8caa6d6

Please sign in to comment.