You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
#3474
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit
- **Documentation**
- Updated the description of the "mixed:N" string option in the training
data arguments to reflect that the "se_atten" descriptor is now
supported only for the Tensorflow backend.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
Copy file name to clipboardExpand all lines: deepmd/utils/argcheck.py
+1-1Lines changed: 1 addition & 1 deletion
Original file line number
Diff line number
Diff line change
@@ -2064,7 +2064,7 @@ def training_data_args(): # ! added by Ziyao: new specification style for data
2064
2064
- int: all {link_sys} use the same batch size.\n\n\
2065
2065
- string "auto": automatically determines the batch size so that the batch_size times the number of atoms in the system is no less than 32.\n\n\
2066
2066
- string "auto:N": automatically determines the batch size so that the batch_size times the number of atoms in the system is no less than N.\n\n\
2067
-
- string "mixed:N": the batch data will be sampled from all systems and merged into a mixed system with the batch size N. Only support the se_atten descriptor.\n\n\
2067
+
- string "mixed:N": the batch data will be sampled from all systems and merged into a mixed system with the batch size N. Only support the se_atten descriptor for Tensorflow backend.\n\n\
2068
2068
If MPI is used, the value should be considered as the batch size per task.'
2069
2069
doc_auto_prob_style='Determine the probability of systems automatically. The method is assigned by this key and can be\n\n\
2070
2070
- "prob_uniform" : the probability all the systems are equal, namely 1.0/self.get_nsystems()\n\n\
0 commit comments