lightning/tests/tests_pytorch/strategies
Adrian Wälchli a0ca2c8bcd
Disable memory sharing on model parameters in ddp-spawn (#18238)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-15 14:39:51 +02:00
..
launchers Disable memory sharing on model parameters in ddp-spawn (#18238) 2023-08-15 14:39:51 +02: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 Re-enable saving and loading model checkpoint in FSDP with PyTorch < 2.0 (#18296) 2023-08-14 11:09:38 -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