![]() * Add test for non-existing mode, the test should fail if something different from `power` or `binsearch` is passed. * Add newline. * Apply fix Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Update tests/tuner/test_scale_batch_size.py * Update pytorch_lightning/tuner/batch_size_scaling.py Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com> |
||
---|---|---|
.. | ||
__init__.py | ||
auto_gpu_select.py | ||
batch_size_scaling.py | ||
lr_finder.py | ||
tuning.py |