-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Closed
Labels
bugSomething isn't workingSomething isn't workingneeds triageWaiting to be triaged by maintainersWaiting to be triaged by maintainersver: 2.5.x
Description
Bug description
The auto_scale_batch_size
arg is not accept in lightning.Trainer
, but accepted in pytorch_lightning.Trainer
.
Error in call to target 'lightning.pytorch.trainer.trainer.Trainer':
TypeError("Trainer.__init__() got an unexpected keyword argument 'auto_scale_batch_size'")
What version are you seeing the problem on?
v2.5
How to reproduce the bug
import lightning as L
L.Trainer(auto_scale_batch_size="binsearch")
Error messages and logs
# Error messages and logs here please
Environment
Current environment
lightning 2.5.0.post0 pypi_0 pypi
lightning-bolts 0.7.0 pypi_0 pypi
lightning-utilities 0.11.9 pypi_0 pypi
pytorch-lightning 1.9.5 pypi_0 pypi
torch 2.5.1 pypi_0 pypi
torchmetrics 1.6.1 pypi_0 pypi
torchvision 0.20.1 pypi_0 pypi
python 3.12.8 h5148396_0
#- OS (e.g., Linux): 22.04.2 LTS
#- CUDA/cuDNN version: CUDA 12.0
#- GPU models and configuration: 8x Quadro RTX 6000
#- How you installed Lightning(`conda`, `pip`, source): pip
More info
No response
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't workingneeds triageWaiting to be triaged by maintainersWaiting to be triaged by maintainersver: 2.5.x