Skip to content

Commit 2bf0769

Browse files
authored
doc: update doc for mixed:N batch size (#3780)
#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 -->
1 parent 9f5116f commit 2bf0769

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/utils/argcheck.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2064,7 +2064,7 @@ def training_data_args(): # ! added by Ziyao: new specification style for data
20642064
- int: all {link_sys} use the same batch size.\n\n\
20652065
- 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\
20662066
- 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\
20682068
If MPI is used, the value should be considered as the batch size per task.'
20692069
doc_auto_prob_style = 'Determine the probability of systems automatically. The method is assigned by this key and can be\n\n\
20702070
- "prob_uniform" : the probability all the systems are equal, namely 1.0/self.get_nsystems()\n\n\

0 commit comments

Comments
 (0)