Carlos Mocholí 67459944ea Avoid FSDP deprecations during save/load with newer torch versions (#19463) ... * Avoid FSDP deprecations during save/load with newer torch versions * Refactor * Tests		2024-02-14 19:43:59 +01:00
..
launchers	CI: enable testing with coming PT 2.2 (#19289)	2024-01-26 16:42:09 +01:00
__init__.py	Rename LightningLite to Fabric (#16244)	2023-01-04 10:57:18 -05:00
test_ddp.py	Rewrite gradient clipping tests (#19262)	2024-01-10 12:39:56 -05:00
test_ddp_integration.py	CI: enable testing with coming PT 2.2 (#19289)	2024-01-26 16:42:09 +01:00
test_deepspeed.py	`set_device` before `init_process_group` (#19184)	2023-12-21 16:28:16 +01:00
test_deepspeed_integration.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00
test_dp.py	Rewrite gradient clipping tests (#19262)	2024-01-10 12:39:56 -05:00
test_fsdp.py	Avoid FSDP deprecations during save/load with newer torch versions (#19463)	2024-02-14 19:43:59 +01:00
test_fsdp_integration.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00
test_registry.py	Support shortcut name for DeepSpeed stage 1 offload (#19075)	2024-02-05 20:53:18 -05:00
test_single_device.py	Rewrite gradient clipping tests (#19262)	2024-01-10 12:39:56 -05:00
test_strategy.py	Drop support for PyTorch 1.12 (#19300)	2024-01-26 11:44:24 -05:00
test_xla.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00
test_xla_fsdp.py	Minor strategy fixes [TPU] (#18774)	2023-10-11 15:26:30 +02:00
test_xla_fsdp_integration.py	Refactor BoringFabric in tests (#19364)	2024-01-30 23:32:45 +01:00