tests/utilities - Lightning-AI/pytorch-lightning

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

0 0 0 Python

	__init__.py
	distributed.py	1.7 KB
	test_all_gather_grad.py	4.9 KB
	test_apply_func_torchtext.py	2.1 KB
	test_apply_func.py	14.0 KB
	test_argparse.py	7.7 KB
	test_auto_restart.py	59.1 KB
	test_cli.py	58.8 KB
	test_cloud_io.py	1.2 KB
	test_data.py	8.3 KB
	test_deepspeed_collate_checkpoint.py	2.3 KB
	test_distributed.py	1.6 KB
	test_dtype_device_mixin.py	3.5 KB
	test_enums.py	2.1 KB
	test_fetching.py	16.8 KB
	test_finite_checks.py	528 B
	test_grads.py	2.6 KB
	test_imports.py	3.3 KB
	test_logger.py	6.5 KB
	test_memory.py	1.8 KB
	test_meta.py	2.4 KB
	test_migration.py	1.4 KB
	test_model_helpers.py	2.5 KB
	test_model_summary.py	12.5 KB
	test_optimizer.py	1.0 KB
	test_parameter_tying.py	2.5 KB
	test_parsing.py	9.4 KB
	test_rank_zero.py	2.6 KB
	test_remote_filesystem.py	3.5 KB
	test_seed.py	3.4 KB
	test_types.py	1.1 KB
	test_upgrade_checkpoint.py	2.0 KB
	test_warnings.py	3.4 KB
	test_xla_device_utils.py	1.8 KB