lightning/tests/strategies
ananthsub 1bd6fc979e
Remove `Strategy.on_tpu` property (#11536)
2022-01-20 08:25:26 +01:00
..
__init__.py Rename training plugin test files & names to strategy (#11303) 2022-01-04 14:32:45 +01:00
test_custom_plugin.py Fix restoring lr scheduler states with deepspeed strategy (#11322) 2022-01-06 12:34:16 +00:00
test_ddp_fully_sharded_with_full_state_dict.py Fix exception message for FSDP running on CPU (#11325) 2022-01-05 18:02:31 +01:00
test_ddp_spawn_strategy.py Remove `Strategy.on_tpu` property (#11536) 2022-01-20 08:25:26 +01:00
test_ddp_strategy.py Remove `Strategy.on_tpu` property (#11536) 2022-01-20 08:25:26 +01:00
test_ddp_strategy_with_comm_hook.py Rename training plugin test files & names to strategy (#11303) 2022-01-04 14:32:45 +01:00
test_deepspeed_strategy.py Use a dataclass as the scheduler config (#11443) 2022-01-18 20:23:32 +01:00
test_sharded_strategy.py Rename training plugin test files & names to strategy (#11303) 2022-01-04 14:32:45 +01:00
test_single_device_strategy.py Remove `Strategy.on_tpu` property (#11536) 2022-01-20 08:25:26 +01:00
test_strategy_registry.py Introduce StrategyRegistry (#11233) 2022-01-05 17:14:18 +05:30
test_tpu_spawn.py Remove `Strategy.on_tpu` property (#11536) 2022-01-20 08:25:26 +01:00