Skip to content

Commit cb49a1f

Browse files
authored
Apply suggestion from @deependujha
1 parent dd9dea6 commit cb49a1f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/callbacks/batch_size_finder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def __init__(
123123
max_trials: int = 25,
124124
batch_arg_name: str = "batch_size",
125125
margin: float = 0.05,
126-
max_val: Optional[int] = None,
126+
max_val: int = 1024,
127127
) -> None:
128128
mode = mode.lower()
129129
if mode not in self.SUPPORTED_MODES:

0 commit comments

Comments
 (0)