tests/tests_pytorch/models - Lightning-AI/pytorch-lightning - Morph

SIGN IN SIGN UP

Lightning-AI / pytorch-lightning UNCLAIMED

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

0 0 104 Python

	conf
	__init__.py
	test_amp.py	5.2 KB
	test_cpu.py	9.3 KB
	test_ddp_fork_amp.py	1.2 KB
	test_fabric_integration.py	2.9 KB
	test_gpu.py	8.4 KB
	test_hooks.py	38.9 KB
	test_hparams.py	37.4 KB
	test_onnx.py	7.9 KB
	test_restore.py	23.5 KB
	test_torch_tensorrt.py	5.2 KB
	test_torchscript.py	9.8 KB
	test_tpu.py	11.0 KB