lightning/pytorch_lightning/tuner
ramonemiliani93 5db832f181
Fix auto scaling mode when calling tune method on trainer. (#7321)
* 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>
2021-05-04 12:03:51 +00:00
..
__init__.py
auto_gpu_select.py Code cleaning in preparation for #7258 [3/n] (#7262) 2021-04-29 14:40:51 +02:00
batch_size_scaling.py Fix auto scaling mode when calling tune method on trainer. (#7321) 2021-05-04 12:03:51 +00:00
lr_finder.py Attach data refactor and tuner bugs [4/n] (#7258) 2021-04-30 13:54:58 +00:00
tuning.py Fix auto scaling mode when calling tune method on trainer. (#7321) 2021-05-04 12:03:51 +00:00