We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7a60d19 commit f2fa81eCopy full SHA for f2fa81e
src/lightning/pytorch/tuner/batch_size_scaling.py
@@ -62,7 +62,7 @@ def _scale_batch_size(
62
63
margin: Margin to reduce the found batch size by to provide a safety buffer. Only applied when using
64
'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.
+ max_val: Maximum batch size limit, to prevent overly large or inefficient batch sizes.
66
67
"""
68
if trainer.fast_dev_run:
0 commit comments