lightning/tests/tests_pytorch/strategies
Rohit Gupta 0886e6352e
Added a check to validate that wrapped FSDP models are used while initializing optimizers (#15301)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2022-11-08 02:10:35 +00:00
..
launchers Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
scripts
__init__.py
test_bagua_strategy.py Prepare CI to run on 3090s (#14910) 2022-09-29 14:01:59 +00:00
test_colossalai.py Grep for potential errors in standalone tests (#15341) 2022-11-05 04:29:38 +01:00
test_common.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_custom_plugin.py
test_ddp.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_ddp_fully_sharded_native.py Added a check to validate that wrapped FSDP models are used while initializing optimizers (#15301) 2022-11-08 02:10:35 +00:00
test_ddp_fully_sharded_with_full_state_dict.py Added a check to validate that wrapped FSDP models are used while initializing optimizers (#15301) 2022-11-08 02:10:35 +00:00
test_ddp_spawn.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_ddp_spawn_strategy.py Align ddp and ddp-spawn strategies in setting up the environment (#11073) 2022-09-29 19:30:09 +02:00
test_ddp_strategy.py Grep for potential errors in standalone tests (#15341) 2022-11-05 04:29:38 +01:00
test_ddp_strategy_with_comm_hook.py Grep for potential errors in standalone tests (#15341) 2022-11-05 04:29:38 +01:00
test_deepspeed_strategy.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_dp.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_hivemind.py
test_registry.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
test_sharded_strategy.py Grep for potential errors in standalone tests (#15341) 2022-11-05 04:29:38 +01:00
test_single_device_strategy.py
test_tpu_spawn.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00