lightning/tests/tests_pytorch/strategies
Adrian Wälchli e8f3863e0f
Add lazy checkpoint loading for FSDP full-state checkpoints (Trainer) (#18379)
2023-08-24 09:35:46 -04:00
..
launchers Improve error message when "if name == main" guard is needed (#18298) 2023-08-18 08:53:01 -04:00
scripts copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
test_common.py True half-precision support in Trainer (#18193) 2023-08-02 07:56:22 -04:00
test_custom_plugin.py Minor miscellaneous fixes (#18068) 2023-07-13 06:01:58 -04:00
test_ddp.py True half-precision support in Trainer (#18193) 2023-08-02 07:56:22 -04:00
test_ddp_spawn.py docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
test_ddp_spawn_strategy.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_ddp_strategy.py docformatter: config with black (#18064) 2023-08-09 10:44:20 -04: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 Cast input before moving to device for all strategies (#18264) 2023-08-09 17:42:55 +02:00
test_fsdp.py Add lazy checkpoint loading for FSDP full-state checkpoints (Trainer) (#18379) 2023-08-24 09:35:46 -04:00
test_registry.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_single_device_strategy.py docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
test_xla.py Allow accessing rank information before processes are launched in XLA (#18194) 2023-07-31 10:37:35 -04:00