Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

auto_scale_batch_size arg not accept by lightning.Trainer #20572

Open
yc-tao opened this issue Feb 3, 2025 · 0 comments
Open

auto_scale_batch_size arg not accept by lightning.Trainer #20572

yc-tao opened this issue Feb 3, 2025 · 0 comments
Labels
bug Something isn't working needs triage Waiting to be triaged by maintainers ver: 2.5.x

Comments

@yc-tao
Copy link

yc-tao commented Feb 3, 2025

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

@yc-tao yc-tao added bug Something isn't working needs triage Waiting to be triaged by maintainers labels Feb 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working needs triage Waiting to be triaged by maintainers ver: 2.5.x
Projects
None yet
Development

No branches or pull requests

1 participant