lightning/tests/tests_pytorch/strategies
Adrian Wälchli 3d41313b7b
Meta device initialization for FSDP in Trainer (#18385)
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2023-08-25 14:18:43 -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_strategy.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_ddp.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_ddp_integration.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_ddp_integration_comm_hook.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_deepspeed.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_fsdp.py Meta device initialization for FSDP in Trainer (#18385) 2023-08-25 14:18:43 -04:00
test_registry.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_single_device.py Organize strategy tests folder (#18377) 2023-08-24 18:01:11 -04:00
test_xla.py Allow accessing rank information before processes are launched in XLA (#18194) 2023-07-31 10:37:35 -04:00