lightning/tests/tests_pytorch/trainer
Quazi Marufur Rahman 2577285dd5
Use LRScheduler for torch >= 1.14 otherwise use _LRScheduler (#15768)
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-12-12 13:33:26 +00:00
..
connectors Fix ddp_spawn -> ddp fallback logic when on LSF cluster (#15657) 2022-11-12 17:26:16 +00:00
dynamic_args Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
flags Set the logger explicitly in tests (#15815) 2022-12-10 13:32:37 +00:00
logging_ Set the logger explicitly in tests (#15815) 2022-12-10 13:32:37 +00:00
optimization Use LRScheduler for torch >= 1.14 otherwise use _LRScheduler (#15768) 2022-12-12 13:33:26 +00:00
properties Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_config_validator.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
test_dataloaders.py Set the logger explicitly in tests (#15815) 2022-12-10 13:32:37 +00:00
test_progress.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_states.py Add `LRFinder` callback (#13802) 2022-10-05 13:15:38 +02:00
test_supporters.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
test_trainer.py Set the logger explicitly in tests (#15815) 2022-12-10 13:32:37 +00:00
test_trainer_cli.py Move accelerator-specific parsing functions with their accelerators (#14753) 2022-09-18 22:48:45 +00:00