Skip to content

Commit 102a2ed

Browse files
A-LOST-WAPITIglwanpre-commit-ci[bot]
authored
doc: fix inconsistency between the docstring and the implementation of argument auto_batch_size of DeepEval with paddle and pytorch backend (#4865)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **New Features** * Automatic batch size handling is now enabled by default when creating a new DeepEval instance. * **Documentation** * Improved formatting consistency in model information and multi-task training guides by removing unnecessary blank lines. <!-- end of auto-generated comment: release notes by coderabbit.ai --> --------- Co-authored-by: glwan <wanguolin@dp.tech> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 1fbe4c1 commit 102a2ed

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

deepmd/dpmodel/infer/deep_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ class DeepEval(DeepEvalBackend):
6262
The output definition of the model.
6363
*args : list
6464
Positional arguments.
65-
auto_batch_size : bool or int or AutomaticBatchSize, default: False
65+
auto_batch_size : bool or int or AutomaticBatchSize, default: True
6666
If True, automatic batch size will be used. If int, it will be used
6767
as the initial batch size.
6868
neighbor_list : ase.neighborlist.NewPrimitiveNeighborList, optional

deepmd/pd/infer/deep_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ class DeepEval(DeepEvalBackend):
5858
The output definition of the model.
5959
*args : list
6060
Positional arguments.
61-
auto_batch_size : bool or int or AutomaticBatchSize, default: False
61+
auto_batch_size : bool or int or AutomaticBatchSize, default: True
6262
If True, automatic batch size will be used. If int, it will be used
6363
as the initial batch size.
6464
neighbor_list : ase.neighborlist.NewPrimitiveNeighborList, optional

deepmd/pt/infer/deep_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ class DeepEval(DeepEvalBackend):
8383
The output definition of the model.
8484
*args : list
8585
Positional arguments.
86-
auto_batch_size : bool or int or AutomaticBatchSize, default: False
86+
auto_batch_size : bool or int or AutomaticBatchSize, default: True
8787
If True, automatic batch size will be used. If int, it will be used
8888
as the initial batch size.
8989
neighbor_list : ase.neighborlist.NewPrimitiveNeighborList, optional

0 commit comments

Comments
 (0)