Skip to content

Commit

Permalink
args: add shuffle_poscar to run jdata
Browse files Browse the repository at this point in the history
  • Loading branch information
njzjz committed Jul 16, 2022
1 parent 32a9996 commit 786b8a7
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions dpgen/generator/arginfo.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@ def model_devi_lmp_args() -> List[Argument]:
doc_model_devi_clean_traj = 'If type of model_devi_clean_traj is bool type then it denote whether to clean traj folders in MD since they are too large. If it is Int type, then the most recent n iterations of traj folders will be retained, others will be removed.'
doc_model_devi_nopbc = 'Assume open boundary condition in MD simulations.'
doc_model_devi_activation_func = 'Set activation functions for models, length of the list should be the same as numb_models, and two elements in the list of string respectively assign activation functions to the embedding and fitting nets within each model. Backward compatibility: the orginal "list of String" format is still supported, where embedding and fitting nets of one model use the same activation function, and the length of the list should be the same as numb_models.'
doc_shuffle_poscar = 'Shuffle atoms of each frame before running simulations.' # TODO: explain the reason to do shuffle?

return [
model_devi_jobs_args(),
Expand Down Expand Up @@ -159,6 +160,7 @@ def model_devi_lmp_args() -> List[Argument]:
doc=doc_model_devi_nopbc),
Argument("model_devi_activation_func", list, optional=True,
doc=doc_model_devi_activation_func),
Argument("shuffle_poscar", bool, optional=False, doc=doc_shuffle_poscar),
]


Expand Down

0 comments on commit 786b8a7

Please sign in to comment.