Skip to content

auto_scale_batch_size arg not accept by lightning.Trainer #20572

@yc-tao

Description

@yc-tao

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

No one assigned

    Labels

    bugSomething isn't workingneeds triageWaiting to be triaged by maintainersver: 2.5.x

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions