lightning/tests/tests_pytorch/strategies
Adrian Wälchli d9493545cf
Allow accessing rank information before processes are launched in XLA (#18194)
2023-07-31 10:37:35 -04:00
..
launchers Use hydra.run.dir (not os.getcwd) for DDP subprocesses' run dir (#18145) 2023-07-30 07:57:47 -04:00
scripts copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
test_common.py Add `Trainer.init_module` and `LightningModule.configure_model` (#18004) 2023-07-14 19:15:05 +02:00
test_custom_plugin.py Minor miscellaneous fixes (#18068) 2023-07-13 06:01:58 -04:00
test_ddp.py Add `Trainer.init_module` and `LightningModule.configure_model` (#18004) 2023-07-14 19:15:05 +02:00
test_ddp_spawn.py Make model test more robust (#18043) 2023-07-10 20:36:18 +00:00
test_ddp_spawn_strategy.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_ddp_strategy.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_ddp_strategy_with_comm_hook.py Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
test_deepspeed_strategy.py DeepSpeed precision simplifications (#18113) 2023-07-20 07:13:31 -04:00
test_fsdp.py Relax the assumption that the root module is FSDP wrapped (#18054) 2023-07-19 15:34:03 +02:00
test_registry.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_single_device_strategy.py Minor miscellaneous fixes (#18068) 2023-07-13 06:01:58 -04:00
test_xla.py Allow accessing rank information before processes are launched in XLA (#18194) 2023-07-31 10:37:35 -04:00