tests/plugins - Lightning-AI/pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

30975 0 0 Python

	environments
	__init__.py
	test_amp_plugins.py	5.4 KB
	test_cluster_integration.py	5.0 KB
	test_custom_plugin.py	1.4 KB
	test_ddp_fully_sharded_with_full_state_dict.py	7.3 KB
	test_ddp_plugin_with_comm_hook.py	4.3 KB
	test_ddp_plugin.py	1.9 KB
	test_ddp_spawn_plugin.py	1.5 KB
	test_deepspeed_plugin.py	23.5 KB
	test_double_plugin.py	5.1 KB
	test_plugins_registry.py	3.5 KB
	test_rpc_plugin.py	2.4 KB
	test_rpc_sequential_plugin.py	6.2 KB
	test_sharded_plugin.py	9.2 KB
	test_single_device_plugin.py	2.2 KB
	test_tpu_spawn.py	3.7 KB