tests/tests_pytorch/utilities - 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

	migration
	testing
	__init__.py
	test_all_gather_grad.py	4.9 KB
	test_auto_restart.py	5.4 KB
	test_combined_loader.py	23.9 KB
	test_compile.py	7.7 KB
	test_consolidate_checkpoint.py	1.3 KB
	test_data.py	14.7 KB
	test_deepspeed_collate_checkpoint.py	2.4 KB
	test_deepspeed_model_summary.py	3.1 KB
	test_dtype_device_mixin.py	1.8 KB
	test_enums.py	934 B
	test_exceptions.py	1.7 KB
	test_grads.py	3.1 KB
	test_imports.py	6.2 KB
	test_memory.py	1.2 KB
	test_model_helpers.py	6.4 KB
	test_model_summary.py	17.8 KB
	test_parameter_tying.py	2.5 KB
	test_parsing.py	8.5 KB
	test_pytree.py	2.4 KB
	test_seed.py	1.7 KB
	test_signature_utils.py	1.1 KB
	test_types.py	1.1 KB
	test_upgrade_checkpoint.py	4.1 KB
	test_warnings.py	2.5 KB