Adrian Wälchli 080eaf38fa Enable setting the sharding strategy as string in FSDP (#18087)		2023-07-15 18:07:09 +02:00
..
launchers	update list of fist party packages (#16859)	2023-03-03 16:55:48 +00:00
__init__.py	Rename LightningLite to Fabric (#16244)	2023-01-04 10:57:18 -05:00
test_ddp.py	Fix state dict loading for ddp/dp in Fabric (#17997)	2023-07-06 13:47:17 +02:00
test_ddp_integration.py	Fix state dict loading for ddp/dp in Fabric (#17997)	2023-07-06 13:47:17 +02:00
test_deepspeed.py	Validate selected device indices in `DeepSpeedStrategy` (#17952)	2023-07-04 18:58:38 +00:00
test_deepspeed_integration.py	Validate selected device indices in `DeepSpeedStrategy` (#17952)	2023-07-04 18:58:38 +00:00
test_dp.py	add & apply flake8-simplify (#17386)	2023-04-24 21:57:08 +00:00
test_fsdp.py	Enable setting the sharding strategy as string in FSDP (#18087)	2023-07-15 18:07:09 +02:00
test_fsdp_integration.py	Support sets for policies in FSDP (#18084)	2023-07-15 17:39:28 +02:00
test_registry.py	[TPU] Rename classes to use XLA instead of TPU (#17383)	2023-04-28 12:36:22 -04:00
test_single_device.py	Support empty weight initialization in `Fabric.init_module()` (#17627)	2023-06-07 18:33:53 +00:00
test_strategy.py	Fix state dict loading for ddp/dp in Fabric (#17997)	2023-07-06 13:47:17 +02:00
test_xla.py	[TPU] For XLA Strategy, added function arg to control `broadcast_master_param()` (#17522)	2023-05-05 17:57:24 +00:00