Skip to content

Commit f2fa81e

Browse files
authored
Apply suggestion from @deependujha
1 parent 7a60d19 commit f2fa81e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/tuner/batch_size_scaling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def _scale_batch_size(
6262
6363
margin: Margin to reduce the found batch size by to provide a safety buffer. Only applied when using
6464
'binsearch' mode. Should be a float between 0 and 1. Defaults to 0.05 (5% reduction).
65-
max_val: Maximum batch size limit. If provided, the found batch size will not exceed this value.
65+
max_val: Maximum batch size limit, to prevent overly large or inefficient batch sizes.
6666
6767
"""
6868
if trainer.fast_dev_run:

0 commit comments

Comments
 (0)