lightning/tests/tests_pytorch
Rohit Gupta 0f6caffa57
Fix deepspeed default precision plugin `amp_level` to O2 (#13897)
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
2022-07-29 20:36:51 +00:00
..
accelerators Cast on host instead of IPU when using `precision=16` (#13880) 2022-07-28 19:26:41 +00:00
benchmarks
callbacks Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
checkpointing
core Support setting the trainer reference recursively for ensembles (#13638) 2022-07-22 19:58:46 +02:00
deprecated_api Cast on host instead of IPU when using `precision=16` (#13880) 2022-07-28 19:26:41 +00:00
helpers Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
lite Support DeepSpeed <0.7.0 (#13859) 2022-07-28 14:38:51 +00:00
loggers Lazy import check for wandb dependency (#13474) 2022-07-22 19:57:46 +02:00
loops Add logging messages to notify when `FitLoop` stopping conditions are met (#9749) 2022-07-23 12:07:47 +00:00
models Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
overrides Support setting the trainer reference recursively for ensembles (#13638) 2022-07-22 19:58:46 +02:00
plugins Fix deepspeed default precision plugin `amp_level` to O2 (#13897) 2022-07-29 20:36:51 +00:00
profilers Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
serve Introduce ServableModuleValidator Callback (#13614) 2022-07-15 11:07:40 -04:00
strategies Run GPU tests with PyTorch 1.12 (#13716) 2022-07-28 19:37:57 +05:30
trainer Run GPU tests with PyTorch 1.12 (#13716) 2022-07-28 19:37:57 +05:30
tuner Fix error handling in learning rate finder (#13845) 2022-07-27 04:32:39 -04:00
utilities Organize model summary utilities (#13893) 2022-07-28 19:23:29 +02:00
__init__.py
conftest.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
run_standalone_tasks.sh Improvements to standalone scripts (#13840) 2022-07-28 23:33:22 +00:00
run_standalone_tests.sh Improvements to standalone scripts (#13840) 2022-07-28 23:33:22 +00:00
test_cli.py Update LightningCLI test for new support in latest release of jsonargparse (#13805) 2022-07-25 09:25:42 +00:00